洛谷P1131 时态同步

题意：

给一个n点的树，每条边都有边权，问从根出发需要增加多少长度，使得最终的儿子到根的距离是一样的

思路：

上来一个思路wa了3次，看完题解之后，又一次豁然开朗……

orz

 #include<bits/stdc++.h>
 using namespace std;
 #define ll long long
 #define il inline
 #define it register int
 #define inf 0x3f3f3f3f
 #define lowbit(x) (x)&(-x)
 #define mem(a,b) memset(a,b,sizeof(a))
 #define mod 1000000007
 const int maxn=5e5+;
 struct node{
     int to,next,w;
 }a[maxn<<];
 int n,s,cnt,tot,head[maxn],maxx[maxn];
 ll ans;
 il void add(int u,int v,ll w){
     a[tot].w=w;a[tot].next=head[u];
     a[tot].to=v;head[u]=tot++;
 }
 void dfs(int u,int qian){
     for(it i=head[u];i!=-;i=a[i].next){
         int v=a[i].to,w=a[i].w;
         if(v==qian){continue;}
         dfs(v,u);
         maxx[u]=max(maxx[u],maxx[v]+w);
     }
     for(it i=head[u];i!=-;i=a[i].next){
         int v=a[i].to,w=a[i].w;
         if(v==qian){continue;}
         ans+=(ll)(maxx[u]-w-maxx[v]);
     }
 }
 int main(){
     scanf("%d%d",&n,&s);
     tot=;mem(head,-);ans=;
     for(it i=;i<n;i++){
         int u,v,w;
         scanf("%d%d%d",&u,&v,&w);
         add(u,v,w);add(v,u,w);
     }
     dfs(s,);
     printf("%lld\n",ans);
     return ;
 }