【CF1605】D. Treelabeling（二分图）

题目链接：https://codeforces.com/contest/1605/problem/D

分析

根据题意，我们需要尽量让两个相连的点之间不能到达，于是想到二分图。
我们发现，两个数若最高位不同，那么必定不能到达，于是可以根据最高位将n个数划分为若干个不同的集合，集合之间的数不能互相到达。（例如$n=9$时，我们可以划分为$4$个集合 { 1 } , { 2 , 3 } , { 4 , 5 , 6 , 7 } , { 8 , 9 } \{1\},\{2,3\},\{4,5,6,7\},\{8,9\} {1},{2,3},{4,5,6,7},{8,9}，集合$size$分别是$1,2,4,2$）
那么我们只需要根据二分图不同颜色的数量将这些集合组合成两个集合，再将答案填入即可。（例如$n=9$，二分图划分出来两种颜色的数量分别是$3,6$，于是就能组合成$1+2=3,4+2=6$）
根据二进制的性质，我们必定可以组合成二分图两个颜色的数量较小的那一个，具体看代码。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
#define fi first
#define se second
#define lson (k << 1)
#define rson (k << 1 | 1)

const ll LINF = 1e18;
const int INF = 1e9 + 7;
const int N = 2e5 + 10;
const int M = 1e6 + 10;
const ll P = 998244353;
const double eps = 1e-7;

int n, sum0, sum1;
int c[N], f[N], ans[N];
vector<int> g[N];

void dfs(int u, int fa, int color)
{
    c[u] = color;
    color == 0 ? sum0++ : sum1++;
    for(int i=0,v;i<g[u].size();i++)
    {
        v = g[u][i];
        if(v == fa) continue;
        dfs(v, u, color ^ 1);
    }
}

int main(){
    int T = 1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++) g[i].clear(), c[i] = 0, f[i] = 0;
        for(int i=1,u,v;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        sum0 = sum1 = 0;
        dfs(1, 0, 0);//二分图划分
        if(sum0 > sum1)
        {
            swap(sum0, sum1);
            for(int i=1;i<=n;i++) c[i] ^= 1;
        }
        for(int i=1;i<=n;i++)
        {
            int hb;
            for(hb=30;hb>=0;hb--) if((i >> hb) & 1) break;
            if((sum0 >> hb) & 1) f[i] = 0;
            else f[i] = 1;
        }
        int cnt0 = 1, cnt1 = 1;
        for(int i=1;i<=n;i++)
        {
            if(c[i] == 0)
            {
                while(f[cnt0] == 1) cnt0++;
                ans[i] = cnt0++;
            }
            else
            {
                while(f[cnt1] == 0) cnt1++;
                ans[i] = cnt1++;
            }
        }
        for(int i=1;i<=n;i++) printf("%d ",ans[i]);
        printf("\n");
    }
    return 0;
}