深度优先遍历

树的重心
原题
B站董晓讲解

任取一点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;
}
posted @ 2024-07-31 20:02  某朝  阅读(29)  评论(0)    收藏  举报