Balancing Act

题目 Balancing Act

找树的重心

树重心的性质：

• 一颗树一定有一个重心。
• 当且仅当图的节点数n为偶数且有一条边连接两个大小均为$n/2$的子树时，树有两个重心，即是这条边的两个端点。
• 每个子树大小都不大于$n/2$
• 重心到所有点的距离和最小

维护两个数组：$size,f$，一个是存储节点个数，一个维护父亲节点

如果当前这个节点的节点数大于$\frac{2}{n}$，则重心一定在这颗子树中，就进行$dfs(v,u,tot)$即可

Code

#include <iostream>
#include <cstdio>
#include <fstream>
#include <cmath>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <algorithm>
#define rep(i,a,b) for(register int (i)=(a);(i)<=(b);(i)++)
using namespace std;
const int maxn=1e5+10;
const int maxm=1e3+10;
int T,n,cnt,m,root;
int in[maxn],size[maxn],f[maxn],head[maxn];

template <class t> inline void read(t &x) {
	x=0;int f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
	while(isdigit(ch)){x=10*x+ch-'0';ch=getchar();}
	x*=f;
}

struct node{
	int v,nex;
}e[maxn<<1];

void add(int u,int v) {
	e[++cnt].v=v;
	e[cnt].nex=head[u];
	head[u]=cnt;
}

void readdata() {
	read(T);
}

void dfs_si(int u,int fa) {
	size[u]=1;
	f[u]=fa;
	for(int i=head[u];i;i=e[i].nex) {
		int v=e[i].v;
		if(v==fa) continue;
		dfs_si(v,u);
		size[u]+=size[v];
	}
}

inline int dfs_ans(int u,int fa,int tot) {
	for(int i=head[u];i;i=e[i].nex) {
		int v=e[i].v;
		if(v==fa) continue;
		if(size[v]*2>tot) return dfs_ans(v,u,tot);
	}
	return u;
}

void init() {
	cnt=0;
	memset(in,false,sizeof(in));
	memset(size,false,sizeof(size));
	memset(e,false,sizeof(e));
	memset(head,false,sizeof(head));
}

void work() {
	while(T--) {
		init();
		read(n);
		rep(i,1,n-1) {
			int x,y;
			read(x),read(y);
			in[y]++;
			add(x,y);add(y,x);
		}
		rep(i,1,n) if(!in[i]) root=i;
		dfs_si(root,0);
		int ans=dfs_ans(root,0,size[root]),ans2=0;
		for(int i=head[ans];i;i=e[i].nex) {
			int v=e[i].v;
			if(v==f[ans]) continue;
			ans2=max(ans2,size[v]);
		}
		ans2=max(ans2,(n-size[ans]));
		printf("%d %d\n",ans,ans2);
	}
}

int main() {
	//freopen("input.txt","r",stdin);
	readdata();
	work();
	return 0;
}