树形DP——BZOJ1060/Luogu1131 [ZJOI2007]时态同步

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;
}