2021牛客暑期多校训练营8 Tree

题目

分析

我们先找出满足基础目标的染色方法。因为基础目标是保证染色后树还联通，所以我们可以对图做一次最小生成树人，然后对这棵树的每一层进行交替染色。这样就满足了基本要求。
然后就是满足额外的要求了。对于额外要求$1$，我们可以从已染色的点中，挑出一些互相独立的点，将他们染成剩下的一种颜色。在选点的时候，注意不要将已染色的点的数量减少到$\frac{n}{3}$以下。
但是有可能基础目标达成后，一种颜色的点的数量大于$\frac{2}{3}n$，这时，我们就可以选择第二个要求。可以证明，此时只要把叶子结点染上另一种颜色即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll T,n,m,c[300500],v[300500],sum[3],leaf[300500];
vector<ll>e[300500];
vector<ll>son[300500];
void dfs(ll x,ll t)
{
    v[x]=1;
    c[x]=t;
    sum[t]++;
    leaf[x]=1;
    for (ll i=0;i<e[x].size();i++)
    if (!v[e[x][i]])
     {
        son[x].push_back(e[x][i]);
        dfs(e[x][i],t^1);
        leaf[x]=0;
     }
}
void dfs2(ll x,ll fa)
{
    for (ll i=0;i<son[x].size();i++)
      dfs2(son[x][i],x);
    if (sum[c[x]]>0&&!v[x])
    {
        v[x]=v[fa]=1;sum[c[x]]--;c[x]=2;sum[c[x]]++;
    }
}
int main()
{
    scanf("%lld",&T);
    while(T--)
    {
        scanf("%lld%lld",&n,&m);
        for (ll i=0;i<=n;i++)
         {
            e[i].clear();son[i].clear();c[i]=-1;v[i]=0;
         }
        sum[0]=sum[1]=sum[2]=0;
        for (ll i=1;i<=m;i++)
        {
            ll x,y;
            scanf("%lld%lld",&x,&y);
            e[x].push_back(y);
            e[y].push_back(x);
        }
        dfs(1,0);
        if (sum[0]>n/3*2||sum[1]>n/3*2)
        {
            for (ll i=1;i<=n;i++)
             if (leaf[i])
              c[i]=2;
        }
        else
        {
            memset(v,0,sizeof(v));
            sum[0]-=n/3;
            sum[1]-=n/3;
            dfs2(1,0);
        }
        for (ll i=1;i<=n;i++)
         if (c[i]==0) printf("R");
         else if (c[i]==1) printf("G");
         else if (c[i]==2) printf("B");
        puts("");
    }
}