846. 樹的重心

給定一顆樹,樹中包含n個結點(編號1~n)和n-1條無向邊。java

請你找到樹的重心,並輸出將重心刪除後,剩餘各個連通塊中點數的最大值。spa

重心定義:重心是指樹中的一個結點,若是將這個點刪除後,剩餘各個連通塊中點數的最大值最小,那麼這個節點被稱爲樹的重心。code

輸入格式

第一行包含整數n,表示樹的結點數。xml

接下來n-1行,每行包含兩個整數a和b,表示點a和點b之間存在一條邊。blog

輸出格式

輸出一個整數m,表示重心的全部的子樹中最大的子樹的結點數目。class

數據範圍

1n10^5
import

輸入樣例

9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

輸出樣例:

4

首先數據範圍10^5,用鄰接表存儲


代碼:
import java.util.Arrays;
import java.util.Scanner;

public class Main{
        static final int N=100005;
        static int h[]=new int[N];
        static int e[]=new int[N*2];//由於是無向邊,idx<=2*N,因此開兩倍空間
        static int ne[]=new int[N*2];
        static boolean vis[]=new boolean[N];
        static int n,idx,ans=N;
        static void add(int a,int b){
                e[idx]=b;
                ne[idx]=h[a];
                h[a]=idx++;
        }
        static int dfs(int u){
                vis[u]=true;
                int sum=1,res=0;
                for(int i=h[u];i!=-1;i=ne[i]){
                        int j=e[i];
                        if(!vis[j]){
                                int s=dfs(j);
                                res=Math.max(res, s);
                                sum+=s;
                        }
                }
                res=Math.max(res, n-sum);
                ans=Math.min(res, ans);
                return sum;
        }
        public static void main(String[] args) {
                Scanner scan=new Scanner(System.in);
                n=scan.nextInt();
                Arrays.fill(h,-1);//初始每一個鄰接表都是空
                for(int i=0;i<n-1;i++){
                        int a=scan.nextInt();
                        int b=scan.nextInt();
                        add(a,b);
                        add(b,a);//無向圖
                }
                dfs(1);
                System.out.println(ans);
        }
}
相關文章
相關標籤/搜索