#bzoj2404#最長鏈(樹的直徑/樹形DP)
阿新 • • 發佈:2019-01-23
2404: 最長鏈
時間限制: 1 Sec 記憶體限制: 128 MB題目描述
給定一棵有n個節點的樹,求每個節點到其他節點的最大距離
輸入
輸入第一行是一個自然數n(n≤10000), 接下來 (n−1) 行描述:
第i行包含兩個自然數 , 表示編號為i的節點連線到的節點編號和這條網線的長度..距離總長不會超過10^9. 每行中的兩個數字用空格隔開.
輸出
輸出包含n行. 第i行表示對於離編號為i的節點最遠的節點與該節點的距離Si(1≤i≤n).
樣例輸入
3
1 1
1 2
樣例輸出
2
3
3
提示
30% n≤100
100% n≤10000
這裡我用樹形dp解,首先預處理
g[i][0]表示以i為根的子樹到其葉子節點的最長距離,g[i][1]表示以i為根的子樹到其葉子節點的次最長距離(g[i][0]和g[i][1]除了起點i以外完全不重合)
f[i][0]表示i的最長鏈(以i為節點),f[i][1]表示i的次最長鏈,接著可以分類討論了:對於i與f[root][0]是否處於同一路徑上(即是否有重合)有:
重合時:
f[i][0]=max( g[i][0], f [fa][1] + W[i] )
f[i][1]=max2( g[i][0], f [fa][1] + W[i], g[i][1])//選擇第二大的
不重合時:
f[i][0]=f [fa][0]+W[i];
f[i][1]=g [fa][0];
初始狀態f[R][0]=g[R][0], f[R][1]=g[R][1]從上向下搜
#include<iostream> #include<cstdio> #include<cstdlib> using namespace std; const int Max=10000; int N,cnt; bool vis[Max+5]; int g[Max+5][2],f[Max+5][2]; int L[(Max<<1)],nxt[(Max<<1)],W[(Max<<1)],fir[(Max<<1)]; void getint(int &num){ char c;int flag=1;num=0; while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1; while(c>='0'&&c<='9'){num=num*10+c-48;c=getchar();} num*=flag; } void add(int a,int b,int v){L[++cnt]=b,nxt[cnt]=fir[a],fir[a]=cnt,W[cnt]=v;} int max(int a,int b){return a<b?b:a;} int max2(int a,int b,int c){ if((a>=b&&a<=c)||(a<=b&&a>=c)) return a; if((b<=a&&b>=c)||(b<=c&&b>=a)) return b; return c; } void Dfs(int r,int fa){ for(int i=fir[r]; i; i=nxt[i])if(L[i]!=fa){ Dfs(L[i],r); g[r][1]=max(g[r][1],g[L[i]][0]+W[i]); if(g[r][1]>g[r][0]) swap(g[r][1],g[r][0]); } } void Dfs1(int r,int fa){ for(int i=fir[r]; i; i=nxt[i])if(L[i]!=fa){ if(f[r][0]-W[i]!=g[L[i]][0]){ f[L[i]][0]=f[r][0]+W[i]; f[L[i]][1]=g[r][0]; } else { f[L[i]][0]=max(g[L[i]][0],f[r][1]+W[i]); f[L[i]][1]=max2(g[L[i]][0],g[L[i]][1],f[r][1]+W[i]); } Dfs1(L[i],r); } } int main(){ getint(N); int b,v,R; for(int i=2; i<=N; ++i){ getint(b),getint(v); add(i,b,v),add(b,i,v); vis[b]=1; } for(R=1; R<=N&&vis[R]; ++R); Dfs(R,0); f[R][0]=g[R][0],f[R][1]=g[R][1]; Dfs1(R,0); for(int i=1; i<=N; ++i) printf("%d\n",f[i][0]); return 0; }