許浩浩

1. #include<bits/stdc++.h>
   
2. using namespace std;
   
3. int n,root;
   
4. int res,finalMaxChildDis;
   
5. struct Node
   
6. {
   
7.     int parent=-1,h=0;
   
8.     int maxChildDis=-1,max=-1,sec=-1;
   
9.     vector<int> child;
   
10. };
    
11. int main()
    
12. {
    
13.     cin.tie(0);
    
14.     cin.sync_with_stdio(0);
    
15. 
    
16.     while(cin>>n)
    
17.     {
    
18.         finalMaxChildDis=-1;
    
19.         Node node[n];
    
20.         for(int i=0;i<n-1;i++)
    
21.         {
    
22.             int p,c;
    
23.             cin>>p>>c;
    
24.             node[c].parent=p;
    
25.             node[p].child.push_back(c);
    
26.         }
    
27.         for(int i=0;i<n;i++)
    
28.         {
    
29.             if(node[i].parent==-1)
    
30.                 root=i;
    
31.             if(node[i].child.size()==0)
    
32.             {
    
33.                 int h=0;
    
34.                 int parent=node[i].parent;
    
35.                 while(parent!=-1)
    
36.                 {
    
37.                     h++;
    
38.                     if(h>node[parent].h)
    
39.                         node[parent].h=h;
    
40.                     else
    
41.                         break;
    
42.                     parent=node[parent].parent;
    
43.                 }
    
44.             }
    
45.             for(int i=0;i<n;i++)
    
46.             {
    
47.                 if(node[i].child.size()>=2)
    
48.                 {
    
49.                     for(int j=0;j<node[i].child.size();j++)
    
50.                     {
    
51.                         int childH=node[node[i].child[j]].h;
    
52.                         if(childH>node[i].max)
    
53.                         {
    
54.                             node[i].sec=node[i].max;
    
55.                             node[i].max=childH;
    
56.                         }
    
57.                         else if(childH>node[i].sec)
    
58.                             node[i].sec=childH;
    
59.                     }
    
60.                     node[i].maxChildDis=node[i].max+node[i].sec+2;
    
61.                     if(node[i].maxChildDis>finalMaxChildDis)
    
62.                         finalMaxChildDis=node[i].maxChildDis;
    
63.                 }
    
64.             }
    
65.         }
    
66.         if(node[root].h>finalMaxChildDis)
    
67.             res=node[root].h;
    
68.         else
    
69.             res=finalMaxChildDis;
    
70.         cout<<res<<endl;
    
71.     }
    
72.     return 0;
    
73. }

複製代碼