换根dp(板子整理)

板子

模板框架，来源于Codeforces群winterzz1

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define pb push_back
const int N=2e5+10;
vector<int>e[N];
int sz[N],n,u,v;
ll dp[N],ans[N];
//视具体情况更改link/cut/leaf函数 
void dp_link(int rt1,int rt2)//使rt1多一个儿子
{
}
void dp_cut(int rt1,int rt2)//从rt1删去一个儿子
{
}
void get_leaf(int rt)//把rt当叶子
{
    sz[rt]=1;
    dp[rt]=1;
}
///-----------------------------
void change_rt(int rt1,int rt2)//把根从rt1切换到rt2 
{
    dp_cut(rt1,rt2);// 使rt1失去儿子rt2 
    dp_link(rt2,rt1);//使rt2加上儿子rt1 
}
void dp_dfs(int x,int fa)
{
    get_leaf(x);
    for(auto &i:e[x])
    {
        if(i!=fa)
        {
            dp_dfs(i,x);
            dp_link(x,i);//回溯时 把x挂到i上 
        }
    }
    return;
}
void dfs(int x,int fa)
{
    ans[x]=dp[x];
    for(auto &i:e[x])
    {
        if(i!=fa)
        {
            change_rt(x,i);//根从x换到i 
            dfs(i,x);
            change_rt(i,x);//回溯 把根换回来 
        }
    }
    return;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;++i)
    {
        scanf("%d%d",&u,&v);
        e[u].pb(v),e[v].pb(u);
    }
    dp_dfs(1,-1);
    dfs(1,-1);
    for(int i=1;i<=n;++i)
    {
        printf("%lld\n",ans[i]);
    }
    return 0;
}