CF Round #670 (Div. 2)-C.Link Cut Centroids //树的重心性质

题目链接

https://codeforces.com/contest/1406/problem/C

题意

给你一棵树( $n$ 个顶点)，现在要求拆掉一条边，再加上一条边使得树的重心唯一

解析

首先需要知道树的重心都有哪些性质：

1. 树中所有点到某个点的距离和中，到重心的距离和是最小的，如果有两个重心，他们的距离和一样。
2. 把两棵树通过一条边相连，新的树的重心在原来两棵树重心的连线上。
3. 一棵树添加或者删除一个节点，树的重心最多只移动一条边的位置。
4. 一棵树最多有两个重心，且相邻。

这样的话，就好办了，如果树的重心只有一个，就选择树上任意一条边删除再加上即可；如果树的重心有两个，就选择其中一个重心作为根，将其最大的子树拆掉即可(我选择将其的一个叶子拆下来，连向该重心)

代码

#include <bits/stdc++.h>
#define ft first
#define sd second
#define pb push_back
#define endl '\n'
#define sz(x) (int)x.size()
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int maxn=1e5+10;
const int inf=0x3f3f3f3f;
int n,t;
vector<int> g[maxn],ok;
int siz[maxn],maxp[maxn],root,ff[maxn],pre[maxn];
//求出树的重心存于ok
void getroot(int u,int fa){
    siz[u]=1;maxp[u]=0;
    for(int i=0;i<sz(g[u]);i++){
        int v=g[u][i];
        if(v==fa)continue;
        getroot(v,u);
        siz[u]+=siz[v];
        maxp[u]=max(maxp[u],siz[v]);
    }
    maxp[u]=max(maxp[u],n-siz[u]);
    if(maxp[u]<maxp[root])ok[0]=u,ok[1]=-1,root=u;
    else if(maxp[u]==maxp[root])ok[1]=u;
}
//记录重心结点最大子树的一个叶子结点及其父亲
void dfs(int u,int fa){
    siz[u]=1;
    ff[u]=-1;
    pre[u]=fa;
    int ma=-1;
    for(int i=0;i<sz(g[u]);i++){
        int v=g[u][i];
        if(v==fa)continue;
        dfs(v,u);
        siz[u]+=siz[v];
        if(siz[v]>ma){
            ma=siz[v];
            ff[u]=v;
        }
    }
}
void solve(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)g[i].clear();
    ok.clear();ok.resize(2,-1);
    for(int i=1,u,v;i<n;i++){
        scanf("%d%d",&u,&v);
        g[u].pb(v);
        g[v].pb(u);
    }
    root=0;
    maxp[0]=inf;
    getroot(1,0);
    if(ok[1]==-1){//只有一个重心
        printf("1 %d\n",g[1][0]);
        printf("1 %d\n",g[1][0]);
        return ;
    }
    dfs(ok[0],0);
    int rot=1;
    for(int i=ff[ok[0]];i!=-1;i=ff[i])rot=i;
    printf("%d %d\n",pre[rot],rot);
    printf("%d %d\n",ok[0],rot);
}
int main() {
    cin>>t;
    while(t--)solve();
    return 0;
}