1. 程式人生 > >luogu1131 [ZJOI2007]時態同步

luogu1131 [ZJOI2007]時態同步

struct 路徑 int span max 葉子 str turn out

num[x]表示x到達葉子最遠路徑。
每個子節點對答案的貢獻是num[x] - (num[t] + edge[i].val)

#include <iostream>
#include <cstdio>
using namespace std;
typedef long long ll;
int n, ss, hea[500005], cnt, uu, vv, ww;
ll num[500005], ans;
struct Edge{
    int too, nxt, val;
}edge[1000005];
void add_edge(int fro, int too, int val){
    edge[++cnt].nxt = hea[fro];
    edge[cnt].too = too;
    edge[cnt].val = val;
    hea[fro] = cnt;
}
void
dfs(int x, int f){ for(int i=hea[x]; i; i=edge[i].nxt){ int t=edge[i].too; if(t!=f){ dfs(t, x); num[x] = max(num[x], num[t]+edge[i].val); } } for(int i=hea[x]; i; i=edge[i].nxt){ int t=edge[i].too; if(t!=f) ans += num[x] - (num[t] + edge[i].val); } } int
main(){ cin>>n>>ss; for(int i=1; i<n; i++){ scanf("%d %d %d", &uu, &vv, &ww); add_edge(uu, vv, ww); add_edge(vv, uu, ww); } dfs(ss, 0); cout<<ans<<endl; return 0; }

luogu1131 [ZJOI2007]時態同步