[C++][算法基础]树的重心(树图DFS)

给定一颗树，树中包含 n 个结点（编号 1∼n）和 n−1 条无向边。

请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。

重心定义：重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中点数的最大值最小，那么这个节点被称为树的重心。

输入格式

第一行包含整数 n，表示树的结点数。

接下来 n−1 行，每行包含两个整数 a 和 b，表示点 a 和点 b 之间存在一条边。

输出格式

输出一个整数 m，表示将重心删除后，剩余各个连通块中点数的最大值。

数据范围

1≤n≤

输入样例

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

输出样例：

4

代码：

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;const int N = 100010;
int StartNode[N],edgeTo[N*2],NextThisNode[N*2];
int idx,n,ans;
int att[N*2];void add(int a,int b){edgeTo[idx] = b;NextThisNode[idx] = StartNode[a];StartNode[a] = idx;idx ++;
}int dfs(int x){att[x] = 1;int sum = 1;int res = 0;for(int i = StartNode[x];i != -1;i = NextThisNode[i]){int j = edgeTo[i];if(att[j] == 0){int temp = dfs(j);res = max(res,temp);sum += temp;}}res = max(n - sum,res);ans = min(res,ans);return sum;
}int main(){int a,b;cin>>n;ans = n;memset(StartNode,-1,sizeof StartNode);for(int i = 0;i < n;i++){cin>>a>>b;add(a,b);add(b,a);}dfs(1);cout<<ans<<endl;return 0;
}