[BZOJ1060][ZJOI2007]时态同步 树形dp

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 #include<cstdlib>
 5 #include<cmath>
 6 #include<algorithm>
 7 #define maxn 500005
 8 using namespace std;
 9 int n,s;
10 struct data {
11     int to,next,val;
12 }e[maxn*2];
13 int head[maxn],cnt;
14 void add(int u,int v,int c){e[cnt].to=v;e[cnt].next=head[u];e[cnt].val=c;head[u]=cnt++;}
15 long long f[maxn];
16 long long ans=0;
17 void dp(int now,int fa) {
18     long long sum=0;
19     for(int i=head[now];i>=0;i=e[i].next) {
20         int to=e[i].to;if(to==fa) continue;
21         dp(to,now);
22         f[now]=max(f[now],e[i].val+f[to]);
23     }
24     for(int i=head[now];i>=0;i=e[i].next) {
25         if(e[i].to==fa) continue;
26         ans+=f[now]-f[e[i].to]-e[i].val;
27     }
28 }
29 int main() {
30     memset(head,-1,sizeof(head));
31     scanf("%d%d",&n,&s);
32     for(int i=1;i<n;i++) {
33         int u,v,c;
34         scanf("%d%d%d",&u,&v,&c);
35         add(u,v,c);add(v,u,c);
36     }
37     dp(s,0);
38     printf("%lld",ans);
39 }