码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ1060

时间:2015-08-27 23:13:29      阅读:263      评论:0      收藏:0      [点我收藏+]

标签:dp   树形dp   bzoj   

传送门:BZOJ1060

有些意思的树形DP。

我只想到了用f(i,j)表示以i为根的子树的权值之和为j的最小花费,但我没有想到这个j居然是可以贪心从而藏进去的……

注意到有以下事实成立:在靠近根的节点使用技能更优秀。

于是贪心即可,我们维护每个结点与其子树中叶子结点的最大距离,然后枚举它的子结点,加上它的最大距离与它子结点的最大距离与该边权值之差即可。

比较坑的是,这题标程统计最大距离时忘开long long,统计答案时又开了long long……

最大距离应该开long long!不开long long 只是因为数据有误无法AC!

最大距离应该开long long!不开long long 只是因为数据有误无法AC!

最大距离应该开long long!不开long long 只是因为数据有误无法AC!

重要的事情说三遍。

代码上的小细节见下。

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;

struct Edge{
    int from,to;
    int w;
    Edge(){
        from=to=0;
        w=0;
    }
    Edge(int from_,int to_,int w_){
        from=from_;
        to=to_;
        w=w_;
    }
};

vector<Edge> G[500005];
int maxn[500005];
int n;
int root;
long long ans;

void AddEdge(int from,int to,int w)
{
    G[from].push_back(Edge(from,to,w));
    G[to].push_back(Edge(to,from,w));
}
void Dfs(int root,int fa)
{
    for(int i=0;i<G[root].size();i++){
        Edge e=G[root][i];
        if(e.to==fa)
            continue;
        Dfs(e.to,root);
        maxn[root]=max(maxn[root],maxn[e.to]+e.w);
    }
    for(int i=0;i<G[root].size();i++){
        Edge e=G[root][i];
        if(e.to==fa)
            continue;
        ans+=maxn[root]-maxn[e.to]-e.w;
    }
}

void Readdata()
{
    freopen("loli.in","r",stdin);
    scanf("%d\n%d",&n,&root);
    for(int i=1;i<n;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        AddEdge(a,b,c);
    }
}

void Close()
{
    fclose(stdin);
    fclose(stdout);
}

int main()
{
    Readdata();
    Dfs(root,-1);
    cout<<ans;
    Close();
    return 0;
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

BZOJ1060

标签:dp   树形dp   bzoj   

原文地址:http://blog.csdn.net/le_ballon_rouge/article/details/48035319

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!