【DFS】Codeforces717E[Paint it really, really dark gray]题解

题目概述

有一棵 $n$ 个节点的树，每个节点有黑色和粉色，刚开始你在 $1$ ，每次经过一个节点（刚开始不算）后就会改变该节点的颜色，求一种方案使得所有节点都变成黑色。

解题报告

感觉上好像是什么很神奇的题目，但实际上就是思考一下方法然后直接DFS就行了……

对于一个节点 $x(x\not =1)$ ，递归处理所有子树，把所有子树都变成黑色，然后我们要想办法不改变子树状态而改变 $x$ 的颜色（如果 $x$ 是粉色），这个很简单，只需要走到 $x$ 的父节点 $fa$ ，走回 $x$ ，然后再走到 $fa$ 就行了。

对于 $1$ 怎么办？其实也很简单，假设 $1$ 的一个儿子是 $son$ ，那么 $1\to son\to 1\to son$ 即可。

示例程序

#include<cstdio>
using namespace std;
const int maxn=200000;

int n,a[maxn+5];
int E,lnk[maxn+5],nxt[maxn*2+5],son[maxn*2+5];

inline void Add(int x,int y) {son[++E]=y;nxt[E]=lnk[x];lnk[x]=E;}
void Dfs(int x,int fa=0)
{
    for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa)
        printf(" %d",son[j]),a[son[j]]^=1,Dfs(son[j],x),printf(" %d",x),a[x]^=1;
    if (x==1) {if (!a[x]) a[x]^=1,printf(" %d %d %d",son[lnk[x]],x,son[lnk[x]]);} else
    if (!a[x]) a[x]^=1,a[fa]^=1,printf(" %d %d",fa,x);
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d",&n);for (int i=1;i<=n;i++) {scanf("%d",&a[i]);if (a[i]==-1) a[i]=0;}
    for (int i=1,x,y;i<n;i++) scanf("%d%d",&x,&y),Add(x,y),Add(y,x);
    printf("1");for (int i=1;i<=n;i++) if (!a[i]) {Dfs(1);break;}
    return putchar('\n'),0;
}