Balancing Act POJ - 1655 树形dp 树的重心

题解

题目大意 给你一个n的顶点n-1条边的无根树 让你找出树的重心 并求出平衡值
树的重心 找到一个点,其所有的子树中最大的子树节点数最少,那么这个点就是这棵树的重心 平衡值为最大子树节点数量

使用树形dp求解 siz[i]表示i的子树的大小 d[i]表示以i为重的平衡值
求解过程中任选一点为树根 siz使用DFS回溯求解siz[i] = 1 + ∑siz[子树]
在DFS过程中可以同时求出d[i] d[i] = max(以i为根时原父节点子树大小, siz[子树])
因为通往父节点的路径只有一条所以 以i为根时原父节点子树大小 = n(所有节点数量) - di

AC代码

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const int MAXN = 2e4 + 10;
int n;
int siz[MAXN], d[MAXN]; //i的子树大小 以i为重心的平衡值
vector<int> a[MAXN];

void DFS(int x, int f)
{
	siz[x] = 1; //自身1
	for (int i = 0; i < a[x].size(); i++)
		if (a[x][i] != f)
		{
			DFS(a[x][i], x);
			siz[x] += siz[a[x][i]];//加上不是父节点的子树大小
			d[x] = max(d[x], siz[a[x][i]]); //计算平衡值
		}
	d[x] = max(d[x], n - siz[x]); //因为向父节点只有一条边 x为重心时父的子树大小为n-siz[x]
}
int main()
{
#ifdef LOCAL
	freopen("C:/input.txt", "r", stdin);
#endif
	int T;
	cin >> T;
	while (T--)
	{
		cin >> n;
		for (int i = 1; i <= n; i++)
			a[i].clear(), d[i] = siz[i] = 0;
		for (int i = 1; i < n; i++)
		{
			int u, v;
			scanf("%d%d", &u, &v);
			a[u].push_back(v);
			a[v].push_back(u);
		}
		DFS(1, 0); //随意选取一个根
		int k = 0, s = INF;
		for (int i = 1; i <= n; i++)
			if (s > d[i]) s = d[i], k = i;
		cout << k << " " << s << endl;
	}

	return 0;
}