2021-04-07做题总结

2021.4.7做题总结

没事写了几道牛客寒假集训营的题目（
1.红和蓝

给你一颗树，问你能否按如下方式进行染色：

你拿到了一棵树，请你给每个顶点染成红色或蓝色。要求：每个红点周围有且仅有一个红点，每个蓝点周围有且仅有一个蓝点。“周围”的定义：某点周围的点指通过邻边直接连接的点。所谓树，即没有自环、重边和回路的无向连通图。

我们应该关注的是叶子结点，叶子结点必须与其父节点有相同的颜色，并且一个节点最多只能有一个子节点是叶子节点（且颜色相同—）。（一对）
dfs1:给每个节点染色，同一个cnt，代表同一种颜色！

void dfs1(int u, int father)
{
    int son = 0;//记录下子节点的数量
    for(int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if(j == father) continue;
        son ++;
        dfs1(j, u);
    }
    if(son == 0 || f[u] == 0) //如果是个叶子节点或者这个节点还没有被染色
    {
        if(f[father] != 0)//被涂过色了，直接返回
        {
            flag = false;
            return;
        }
        f[u] = f[father] = cnt ++;
    }
}

第二个dfs:相同的cnt，进行染色，注意是交替染色（
在此之前先初始化一个颜色（
col[1] = 1;

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