http://www.lydsy.com/JudgeOnline/problem.php?id=1060
https://www.luogu.org/problem/show?pid=1131
怎么说呢,看懂了明白了以后呢就变得很简单了
题目给定了根,因此我们只需要从根开始遍历和DP
我们可以定义状态:f[i]表示从第i个节点到叶子结点的最长距离
叶子结点f[i]为0
其他结点的状态转移方程是:
f[i]=max(f[j]+w[i][j])(i表示本节点j表示该节点的子节点)
最后的答案其实就是在把从各节点(设为i)出发所有边权改为f[i]的代价(改到最大嘛)
因此答案就是
∑各节点把各边权修改为f[i]的代价
修改代价的计算方法是
f[i]-f[j]-w[i][j](i表示本节点j表示该节点的子节点)
所有代价累加就是我们要输出的东西啦
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll ans=0;int n,rt,f[1000001]={0};
int nedge=0,p[1000001],c[1000001],nex[1000001],head[1000001];
inline void addedge(int x,int y,int z){
p[++nedge]=y;c[nedge]=z;
nex[nedge]=head[x];head[x]=nedge;
}
inline void dfs(int x,int fa){
for(int k=head[x];k;k=nex[k])if(p[k]!=fa){
dfs(p[k],x);
f[x]=max(f[x],f[p[k]]+c[k]);
}
for(int k=head[x];k;k=nex[k])if(p[k]!=fa)ans+=f[x]-f[p[k]]-c[k];
}
int main()
{
scanf("%d%d",&n,&rt);
for(int i=1;i<n;i++){
int x,y,z;scanf("%d%d%d",&x,&y,&z);
addedge(x,y,z);addedge(y,x,z);
}
dfs(rt,0);
printf("%lld",ans);
return 0;
}