「CodeForces - 717E」Paint it really, really dark gray （dfs）

BUPT 2017 summer training (for 16) #1H

题意

每个节点是黑色or白色，经过一个节点就会改变它的颜色，一开始在1节点。求一条路径使得所有点变成黑色。

题解

dfs时每个节点的孩子处理完，这时候如果颜色是白色，那么再去一下父亲节点再回来，就变成黑色了。

如果是1号点，那就去它的孩子节点，再回来，再去它孩子节点。

代码

#include <cstdio>
#define N 200005
struct edge{
int to,next;
}e[N<<1];
int head
,cnt;
int n,a
;
void add(int u,int v){
e[++cnt]=(edge){v,head[u]};
head[u]=cnt;
}
int ans[N<<1],top;
void dfs(int x,int fa){
ans[++top]=x;
a[x]*=-1;
for(int i=head[x];i;i=e[i].next){
int v=e[i].to;
if(v==fa)continue;
dfs(v,x);
ans[++top]=x;
a[x]*=-1;
}
if(a[x]==-1){
if(x!=1){
ans[++top]=fa;
a[fa]*=-1;
ans[++top]=x;
a[x]=1;
}else{
ans[++top]=e[head[1]].to;
ans[++top]=1;
ans[++top]=e[head[1]].to;
}
}
}
int main(){
scanf("%d",&n);
bool ex=0;
for(int i=1;i<=n;++i){
scanf("%d",&a[i]);
if(a[i]==-1)
ex=1;
}
for(int i=1,u,v;i<n;++i){
scanf("%d%d",&u,&v);
add(u,v);add(v,u);
}
if(!ex)printf("1");
else{
a[1]*=-1;
dfs(1,0);
for(int i=1;i<=top;++i)
printf("%d ",ans[i]);
}
return 0;
}