bzoj1060: [ZJOI2007]時態同步

bzoj1060: [ZJOI2007]時態同步
有生之年可以見到一道裸的樹DP真是幸運。
A了以後翻狀態忽然發現有個代碼長一樣的，看了一眼是隔壁ljss神犇……

---

題解

發現不會寫題解
mxlen[i]記錄以i爲根子樹所需的時間。dfs即可。

---

代碼

#include<iostream>
#include<cstdio>
#define mxn 500005
using namespace std;
int n,s,a,b,t;
long long mxlen[mxn],ans;
int to[mxn<<1],hd[mxn<<1],lk[mxn],len[mxn<<1],cnt;
void add(int u,int v,int w)
{to[++cnt]=v,len[cnt]=w,hd[cnt]=lk[u],lk[u]=cnt;}
void dfs(int x,int last)
{
    for(int i=lk[x];i;i=hd[i])
    if(to[i]!=last)
    dfs(to[i],x),mxlen[x]=max(mxlen[to[i]]+len[i],mxlen[x]);
    for(int i=lk[x];i;i=hd[i])
    if(to[i]!=last)
    ans+=mxlen[x]-mxlen[to[i]]-len[i];
}
int main()
{
    scanf("%d%d",&n,&s);
    for(int i=1;i<n;i++)
    scanf("%d%d%d",&a,&b,&t),add(a,b,t),add(b,a,t);
    dfs(s,0);
    printf("%lld",ans);
}