DP求树的重心

时间:2023-03-08 15:51:28
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
const int maxx = +;
vector<int>G[maxx];
int d[maxx];//保存的是当前节点的儿子节点数目
int vis[maxx];
int sum,num,n;
void dfs(int x)
{
vis[x]=;
int num_son=;
for (int i=; i<G[x].size(); i++)
{
int now=G[x][i];//当前访问的节点
if(!vis[now]) //如果没有访问
{
dfs(now);//DFS
d[x]+=(d[now]+);//D[now]保存的是now的儿子节点个数,+1就是当前节点的儿子节点个数
num_son=max(d[now]+,num_son);//取大的
}
}
num_son=max(num_son,n-d[x]-);
if (num_son<sum || (num_son==sum && x<num))//维护
{
sum=num_son;
num=x;
}
return ;
}
int main()
{
int t,u,v;
scanf("%d",&t);
while(t--)
{
memset(vis,,sizeof(vis));
memset(d,,sizeof(d));
for (int i=; i<=n; i++)
{
G[i].clear();
}
memset(vis,,sizeof(vis));
scanf("%d",&n);
sum=n;
num=;
for (int i=; i<n; i++)
{
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs();
printf("%d %d\n",num,sum);
}
return ;
}