bzoj1060: [ZJOI2007]时态同步

题解

代码

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