黃暐鈞

1. #include<bits/stdc++.h>
   
2. using namespace std;
   
3. int n;//人數
   
4. int root;//根節點
   
5. int res;
   
6. int finalMaxChildDis;
   
7. struct Node
   
8. {
   
9.     int parent=-1,h=0;
   
10.     int maxChildDis=0,max=-1,sec=-1;
    
11.     vector<int> child;
    
12. };
    
13. int main()
    
14. {
    
15.         cin>>n;
    
16.         int finalMaxChildDis=-1;
    
17.         Node node[n];//node資料庫_共n人
    
18.         for(int i=0;i<n-1;i++)//n個人有n-1個連結
    
19.         {
    
20.             int p;//父節點
    
21.             int c;//子節點
    
22.             cin>>p>>c;
    
23.             node[c].parent=p;//c屬於b->p屬於node[c].parent
    
24.             node[p].child.push_back(c);//一個父節點可以同時有很多子節點->用vector存
    
25.         }//建立關係
    
26. 
    
27.         for(int i=0;i<n;i++)//n次
    
28.         {
    
29.             if(node[i].parent==-1)//該節點沒有設定父節點->最高->根節點
    
30.             {
    
31.                 root=i;
    
32.             }
    
33.             if(node[i].child.size()==0)//確認該節點的子節點數量:為0->最底部
    
34.             {
    
35.                 int h=0;
    
36.                 int parent=node[i].parent;//額外定義 不影響node資料
    
37.                 while(parent!=-1)//有父節點
    
38.                 {
    
39.                     h++;
    
40.                     if(h>node[parent].h)
    
41.                     {
    
42.                         node[parent].h=h;//找最深
    
43.                     }
    
44.                     else
    
45.                     {
    
46.                         break;
    
47.                     }
    
48.                     parent=node[parent].parent;//繼續往上找
    
49.                 }
    
50.             }
    
51.         }
    
52.         for(int i=0;i<n;i++)//n次
    
53.         {
    
54.             if(node[i].child.size()>=2)//子節點數量>=2
    
55.             {
    
56.                 for(int j=0;j<node[i].child.size();j++)//重複'子節點數量'次
    
57.                 {
    
58.                     int childH=node[node[i].child[j]].h;//取出子節點的深度
    
59.                     if(childH>node[i].max)//找最深
    
60.                     {
    
61.                         node[i].sec=node[i].max;//找第二深(可能介於兩個最大之間)
    
62.                         node[i].max=childH;
    
63.                     }
    
64.                     else if(childH>node[i].sec)//介於兩個最大之間->更新第二大
    
65.                     {
    
66.                         node[i].sec=childH;
    
67.                     }
    
68.                 }
    
69.                 node[i].maxChildDis=node[i].max+node[i].sec+2;//計算總距離
    
70.                 if(node[i].maxChildDis>finalMaxChildDis)//比較並更新
    
71.                 {
    
72.                     finalMaxChildDis=node[i].maxChildDis;
    
73.                 }
    
74.             }
    
75.         }
    
76.         if(node[root].h>finalMaxChildDis)//根節點距離最遠 不用找另外的子節點距離and+2
    
77.         {
    
78.             res=node[root].h;
    
79.         }
    
80.         else
    
81.         {
    
82.             res=finalMaxChildDis;
    
83.         }
    
84.     cout<<res;
    
85. }

複製代碼