AT_arc161_c 题解

一、题目描述：

　　给你一颗 $n$ 个点的树，每个点的颜色要么是 $W$ ，要么是 $B$ 。满足每个点的度数是奇数。

　　将这 $n$ 个点同时进行染色，这个点被染成它所连接的点中占多数的颜色。给你染色之后的颜色序列。

　　求是否存在一种颜色序列，使得染色之后可以变成现在的颜色序列。

　　如果不存在，输出 $-1$ ；如果存在，输出这个颜色序列。$1\le n\le 2\times {10}^5$ 。

---

 二、解题思路：

　　$dfs$ 以随意一个点为根建树。

　　设染色后的 $i$ 颜色序列为 $c_i$ 。

　　设染色前的 $i$ 颜色序列为 $an{s}_i$ 。

　　我们将 $ans$ 数组初始化为 $-1$ 。

　　然后暴力统计即可，注意两个点：

　　$1.\mathrm{当}\mathrm{一}\mathrm{个}\mathrm{点}i\mathrm{的}\mathrm{儿}\mathrm{子}\mathrm{不}\mathrm{需}\mathrm{要}\mathrm{它}\mathrm{时}，\mathrm{贪}\mathrm{心}\mathrm{地}\mathrm{将}an{s}_i\mathrm{赋}\mathrm{值}\mathrm{为}{c}_{fa}。$

　　$2.\mathrm{注}\mathrm{意}\mathrm{根}\mathrm{节}\mathrm{点}\mathrm{只}\mathrm{有}\mathrm{儿}\mathrm{子}，\mathrm{没}\mathrm{有}\mathrm{父}\mathrm{亲}。$

　　没什么好讲的，因为本来这个题就不难。

---

 三、完整代码：

#include<iostream>
#define N 200010
using namespace std;
string s;
int T,n,u1,v1;
int c[N],du[N],tot[N],ans[N];
struct EDGE{
    int v,nxt;
}edge[N*2];
int head[N],cnt;
void add(int u,int v)
{
    edge[++cnt].v=v;
    edge[cnt].nxt=head[u];
    head[u]=cnt;
}
int dfs(int u,int ff)
{
    for(int i=head[u];i!=-1;i=edge[i].nxt)
        if(edge[i].v!=ff)
        {
            if(!dfs(edge[i].v,u))
                return 0;
            if(ans[edge[i].v]==-1)
                ans[edge[i].v]=c[u];
            if(ans[edge[i].v]==c[u])
                tot[u]++;
        }
    if(tot[u]*2<du[u])
    {
        if(ans[ff]==-1)    ans[ff]=c[u];
        if(!ff||tot[u]*2+2<du[u]||ans[ff]!=c[u])
            return 0;
    }
    return 1;
}
void solve()
{
    cin>>n;cnt=0;
    for(int i=1;i<=n;i++)
        head[i]=ans[i]=-1,du[i]=tot[i]=0;
    for(int i=1;i<n;i++)
    {
        cin>>u1>>v1;
        add(u1,v1),du[u1]++;
        add(v1,u1),du[v1]++;
    }
    cin>>s;
    for(int i=0;i<n;i++)
        c[i+1]=s[i]=='W';
    if(!dfs(1,0))
    {
        cout<<-1<<'\n';
        return ;
    }
    for(int i=1;i<=n;i++)
        ans[i]?cout<<"W":cout<<"B";
    cout<<'\n';
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>T;
    for(int i=1;i<=T;i++)
        solve();
    return 0;
}

---

四、写题心得：

　　我突然发现这根本就不是什么心得，而是我的吐槽。

　　这次 $arc$ 感觉比以前要简单一些（也有可能是我变强了$qwq$），但是我忘了这场比赛，导致迟到了！

　　所以是 $unrated$ ，写起来一点动力都没有，$C$ 题以为把题理解错了一直没写出来，还耽搁了特别水的 $D$ 题。

　　不过问题不大。加油吧！拜拜！