传送门:BZOJ1060
有些意思的树形DP。
我只想到了用
注意到有以下事实成立:在靠近根的节点使用技能更优秀。
于是贪心即可,我们维护每个结点与其子树中叶子结点的最大距离,然后枚举它的子结点,加上它的最大距离与它子结点的最大距离与该边权值之差即可。
比较坑的是,这题标程统计最大距离时忘开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;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
原文地址:http://blog.csdn.net/le_ballon_rouge/article/details/48035319