cqyz oj |【训练题】树的换根操作 | 模板题

Description

我们知道：N个点的无向图，满足下列三个条件中任意两个，那么就是一棵无根树：
(1)、连通；
(2)、有N-1条边；
(3)、任意两点有且仅有一条路径
我们也知道，一棵无根树中，任意一个点都可以作为根。现在的问题是，给出一棵含N个结点的无根树（结点编号为1..N），然后给出若干查询：query x y，其含义为：当以x为根时，y的父亲是谁（若无父亲，则输出0,，和以y为根的子树结点数量。

Input

第1行：包含两个整数N,M，分别表示树的结点数量和查询数量；　　接下来的N-1行，每行包含两个整数x,y，表示x和y之间有一条边连接；　　再接下来的M行，每行一条查询命令。

Output

包含M行，每行输出两个整数，表示查询结果

Sample Input 1

5 3
1 2
3 1
4 3
3 5
query 1 5
query 4 1
query 5 3

Sample Output 1

3 1
3 2
5 4

Hint

1<=n<=50000

核心换根操作，直接上代码

void flip(int x){//把x换成根 
    if(fa[x]){//若不是真正的树根 
        int y=fa[x];
        flip(y);//先把fa[x]换成根 
        //再把x换成根
        fa[x]=0;//默认根的父亲节点为0 
        fa[y]=x;
        siz[y]=n-siz[x];//换根后树y的节点数为除了树x上的点的数量 
        siz[x]=n; 
    }
}

该算法时间复杂度O(d)对于极端数据depth(x)=n时,和直接重新生成一棵以x为根的树O(n)时间复杂度相同，但是对于随机数据通常快得多。

完整代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define inf 0x3f3f3f3f
#define maxn 50005
#define maxm 100005
#define _rep(i,a,b) for(int i=(a);i<=(b);++i)
#define _per(i,a,b) for(int i=(a);i>=(b);--i)
using namespace std;
typedef long long ll;
int fir[maxm],ne[maxm],to[maxm],np=0;
void add(int x,int y){
    ne[++np]=fir[x];
    fir[x]=np;
    to[np]=y;
}

int fa[maxn],siz[maxn];
int ans=0;
void dfs(int u){
    siz[u]=1;
    for(int i=fir[u];i;i=ne[i]){
        int v=to[i];
        if(v==fa[u])continue;
        
        fa[v]=u;
        dfs(v);
        siz[u]+=siz[v];
    }
}

int n,m;
void flip(int x){//把x换成根 
    if(fa[x]){//若不是真正的树根 
        int y=fa[x];
        flip(y);//先把fa[x]换成根 
        //再把x换成根
        fa[x]=0;//默认根的父亲节点为0 
        fa[y]=x;
        siz[y]=n-siz[x];//换根后树y的节点数为除了树x上的点的数量 
        siz[x]=n; 
    }
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1,x,y;i<n;i++){
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    fa[1]=0;
    dfs(1);
    
    char op[10];
    int x,y;
    while(m--){
        scanf("%s%d%d",op,&x,&y);
        flip(x);
        printf("%d %d\n",fa[y],siz[y]);
    }
    return 0;
}

cqyz oj |【训练题】树的换根操作 | 模板题

原文：https://www.cnblogs.com/de-compass/p/11239383.html