深度优先遍历
任取一点u,若以u为重心,则分为两类:一类是u的子树,一类是u上面的部分
需要算出u的最大子树的节点树和u上面的部分的节点数,然后取两者的最大值。
add函数:a所对应的单链表中插入b a作为根
我的代码
#include<iostream>
using namespace std;
const int N = 100010;
int h[N], e[2 * N], ne[2 * N], idx;
bool vis[N];
int n, ans = N;
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
int dfs(int u) {
vis[u] = true;
int size = 0, sum = 1;
for (int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if (vis[j])continue;
int s = dfs(j);
size = max(size, s);
sum += s;
}
ans = min(ans,max(size,n-sum));
return sum;
}
int main() {
memset(h, -1, sizeof(h));
cin >> n;
for (int i = 1; i < n; i++)
{
int a, b;
cin >> a >> b;
add(a, b); add(b, a);
}
dfs(1);
cout << ans << endl;
}