HDU 2196 Computer

標籤：style   blog   color   os   io   for   

第一次dfs求出每個點的最大和次大長度，由下向上更新

第二次由上向下更新。第二次dfs是為了處理某個點，最大長度不是向子節點延伸的長度，而是從父親節點過來的

 

  1 #include<iostream>  2 #include<cstdio>  3 #include<cstring>  4 #include<string>  5 #include<set>  6 #include<vector>  7 #include<map>  8 #include<algorithm>  9 #include<cmath> 10 #include<stack> 11 #include<stdlib.h> 12 using namespace std; 13 #define mmax 10000+10 14 int F_max[mmax],S_max[mmax],F_maxid[mmax],S_maxid[mmax]; 15 const int MAXN=10010; 16 int tol; 17 struct Node 18 { 19     int to; 20     int next; 21     int len; 22 }edge[MAXN*2]; 23 int head[MAXN]; 24 void add(int a,int b,int len) 25 { 26     edge[tol].to=b; 27     edge[tol].len=len; 28     edge[tol].next=head[a]; 29     head[a]=tol++; 30     edge[tol].to=a; 31     edge[tol].len=len; 32     edge[tol].next=head[b]; 33     head[b]=tol++; 34 } 35 int n; 36 void dfs1(int pos,int fa) 37 { 38    //cout<<"spo="<<pos<<endl; 39     F_max[pos]=S_max[pos]=0; 40     F_maxid[pos]=S_maxid[pos]=0; 41     for(int i=head[pos]; i!=-1; i=edge[i].next){ 42         int to=edge[i].to; 43         if(fa==to) 44             continue; 45         int llen=edge[i].len; 46         dfs1(to,pos); 47         if(S_max[pos]<F_max[to]+llen){ 48             S_max[pos]=F_max[to]+llen; 49             S_maxid[pos]=to; 50             if(S_max[pos]>F_max[pos]){ 51                 swap(S_max[pos],F_max[pos]); 52                 swap(S_maxid[pos],F_maxid[pos]); 53             } 54         } 55     } 56  57 } 58 void dfs2(int pos,int fa) 59 { 60     for(int i=head[pos]; i!=-1; i=edge[i].next){ 61         int to=edge[i].to; 62         int llen=edge[i].len; 63         if(to==fa) 64             continue; 65         if(to==F_maxid[pos]){ 66             if(S_max[to]<S_max[pos]+llen){ 67                 S_max[to]=S_max[pos]+llen; 68                 S_maxid[to]=pos; 69                 if(S_max[to]>F_max[to]){ 70                 swap(S_max[to],F_max[to]); 71                 swap(S_maxid[to],F_maxid[to]); 72                 } 73             } 74         } 75         else{ 76              if(S_max[to]<F_max[pos]+llen){ 77                 S_max[to]=F_max[pos]+llen; 78                 S_maxid[to]=pos; 79                 if(S_max[to]>F_max[to]){ 80                 swap(S_max[to],F_max[to]); 81                 swap(S_maxid[to],F_maxid[to]); 82                 } 83             } 84         } 85         dfs2(to,pos); 86     } 87 } 88 void solve() 89 { 90     dfs1(1,0); 91     dfs2(1,0); 92     for(int i=1;i<=n;i++) 93         cout<<F_max[i]<<endl; 94 } 95 int main() 96 { 97     while(cin>>n) 98     { 99         memset(head,-1,sizeof(head));100         tol=0;101         for(int i=1; i<n; i++){102             int a,b;103             cin>>a>>b;104             add(i+1,a,b);105         }106         solve();107     }108 }