C. Link Cut Centroids 【树的重心】1700

C. Link Cut Centroids

图片说明
图片说明
图片说明

思路

把一个重心下的子树上的一个叶子节点,移动到另一个重点作为孩子节点即可

代码

#include <bits/stdc++.h>
#define ios ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define debug_in  freopen("in.txt","r",stdin)
#define debug_out freopen("out.txt","w",stdout);
#define pb push_back
#define all(x) x.begin(),x.end()
#define fs first
#define sc second
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> pii;
const ll maxn = 1e6+10;
const ll maxM = 1e6+10;
const ll inf = 1e8;
const ll inf2 = 1e17;

template<class T>void read(T &x){
    T s=0,w=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
    x = s*w;
}
template<class H, class... T> void read(H& h, T&... t) {
    read(h);
    read(t...);
}

template<class T> void pt(T x){
    cout<<x<<" ";
}

template <typename T, typename... Args>
void pt(const T& t, const Args& ... data){
    cout<<t<<" ";
    pt(data...);
}


//--------------------------------------------

int T,N;
int h[maxn],e[maxn],ne[maxn],idx = 1;
int sz[maxn];
void add(int a,int b){
    e[++idx] = b;
    ne[idx] = h[a];
    h[a] = idx;
}
int H1,H2,mx = inf;
int yz,yzfa;
void dfs(int u,int fa){
    sz[u] = 1;
    int curmx = 0;
    for(int i = h[u];i; i = ne[i]){
        int v = e[i];
        if(v == fa) continue;
        dfs(v,u);
        sz[u] += sz[v];
        curmx = max(curmx,sz[v]);
    }
    curmx = max(curmx,N-sz[u]);
    if(curmx < mx){
        mx = curmx;
        H1 = u;
        H2 = -1;
    }else if(curmx == mx){
        H2 = u;
    }
}
void dfs2(int u,int fa){
    int yy = 1;
    for(int i = h[u];i;i = ne[i]){
        int v = e[i];
        if(v == fa || v == H2) continue;
        yy = 0;
        dfs2(v,u);
    }
    if(yy){
        yz = u;
        yzfa = fa;
    }
}
void solve(){
    if(H2 == -1){
        int v = -1;
        for(int i = h[H1];i;i = ne[i]){
            v = e[i]; break;
        }
        printf("%d %d\n",v,H1);
        printf("%d %d\n",v,H1);
    }else{
        dfs2(H1,H2);
        printf("%d %d\n",yz,yzfa);
        printf("%d %d\n",yz,H2);
    }
}
int main(){
    // debug_in;

    read(T);
    while(T--){
        idx = 1,mx = inf;
        read(N);
        for(int i = 1;i<=N;i++) h[i] = 0,sz[i] = 0;
        for(int i = 1;i<=N-1;i++){
            int a,b;read(a,b);
            add(a,b);
            add(b,a);
        }
        dfs(1,-1);
        solve();
    }



    return 0;
}
全部评论

相关推荐

工作基本定了,比较满意。这深度学习真是难搞啊,论文看不懂,代码看不懂,不会改,创新点???&nbsp;还有一个月中期,怎么搞,已经抑郁了,没有小论文要求,我真是非常幸运了,不然就我这科研能力…………&nbsp;感觉自己对自己失去信心,无时无刻不在否定自己取得一些成绩都是运气好罢了,真的能顺利毕业吗?严重内耗、无力感爆棚啊…………
AI牛可乐:哎呀,听起来你最近压力好大呀,毕业论文和工作的双重压力一定让你感觉像是背了两座山呢。深度学习确实挺复杂的,就像是要在一片迷雾中找到出路一样。但是,你已经很棒了,工作都定好了,这是很大的成就呢! 论文的事情,别太担心。每个人都有看不懂代码和论文的时候,这是正常的。至于创新点嘛,有时候灵感就像小猫一样,需要一点一点地哄出来的。你可以先试试把你看过的资料整理一下,说不定就能找到新的思路呢。 中期答辩前,你可以列个计划,一步一步来,把大问题分解成小问题,一点一点克服。记得,你不是一个人在战斗,你的导师、同学都是你的战友哦。 至于信心嘛,别否定自己,每个人都会有低谷的时候。你已经取得的成绩,都是你实力的体现。毕业不是问题,只要你不放弃,就一定能顺利走过去的。加油哦,我相信你可以的! 对了,如果你愿意的话,可以点击我的头像,我们私信聊,也许我能帮你出出主意,或者就是简单地聊聊天,分散一下你的注意力也好呀。🐮💪🌟
点赞 评论 收藏
分享
头像
10-09 19:35
门头沟学院 Java
洛必不可达:java的竞争激烈程度是其他任何岗位的10到20倍
点赞 评论 收藏
分享
10-18 13:01
已编辑
西安理工大学 C++
小米内推大使:建议技能还是放上面吧,hr和技术面试官第一眼想看的应该是技能点和他们岗位是否匹配
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务