题目 Balancing Act
找树的重心
树重心的性质:
- 一颗树一定有一个重心。
- 当且仅当图的节点数n为偶数且有一条边连接两个大小均为 n / 2 n/2 n/2的子树时,树有两个重心,即是这条边的两个端点。
- 每个子树大小都不大于 n / 2 n/2 n/2
- 重心到所有点的距离和最小
维护两个数组: s i z e , f size,f size,f,一个是存储节点个数,一个维护父亲节点
如果当前这个节点的节点数大于 2 n \frac{2}{n} n2,则重心一定在这颗子树中,就进行 d f s ( v , u , t o t ) dfs(v,u,tot) 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;
}