牛客寒假训练 红和蓝

题目在这

考虑叶结点，因为叶结点必须和一个结点同色，因此只能是他的父节点。
因此对于叶结点，和他父亲染相同的颜色， 两个结点构成了一个组合 。如果出现两个子节点都要和父节点染相同的颜色，即都要和父节点构成组合，则不成立。
如果对于某个结点u，遍历其子节点后其仍没有被染色，说明其子节点已经和子节点的子节点构成组合,因此,u必须和他的父节点同色。
最后特判根节点能不能被染色。

#include <iostream>
#include <cstring>

using namespace std;
const int N = 2e5 + 10;
int h[N], e[N], ne[N], idx;
int col[N], st[N], cnt, success;

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

void dfs1(int u, int fa)
{
    int son = 0;
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j == fa)    continue;
        son ++;
        dfs1(j, u);
    }
    if (son == 0 || !col[u])
    {
        if (col[fa])
        {
            success = 0;
            return;
        }
        col[fa] = col[u] = ++ cnt;
    }
}

void dfs2(int u, int fa)
{
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j == fa)    continue;
        if (col[j] == col[u])
            st[j] = st[u];
        else    st[j] = st[u] ^ 1;
        dfs2(j, u);
    }
}

int main()
{
    memset(h, -1, sizeof h);
    success = 1;
    int n;
    cin >> n;
    for (int i = 0; i < n - 1; i ++)
    {
        int x, y;
        scanf("%d%d", &x, &y);
        add(x, y), add(y, x);
    }
    
    dfs1(1, 0);
    if (!success || col[0])
    {
        cout << -1;
        return 0;
    }
    
    dfs2(1, -1);
    for (int i = 1; i <= n; i ++)
        if (st[i])    cout <<'R';
        else    cout << 'B';
    
}