CF 717E[dfs]

题意:给定一棵树,每个节点初始有颜色,共两种颜色.你一开始在点1,走到哪个点就把它的颜色改变.输出一种合法的方案使最终的树颜色都变为 1 <script type="math/tex" id="MathJax-Element-105">1</script>.

dfs构造
对每个节点如果当前为-1就从它到父亲节点走一遍
如果当前是root为-1没有父亲就随便找个儿子走一遍再走到那个儿子上(因为是dfs处理它的儿子都已经是1了)

#include<cstdio>
#define N 200005
using namespace std;

inline char nc() {
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &x) {
    char c=nc(),b=1;
    for(;!(c>='0'&&c<='9');c=nc()) if(c=='-') b=-1;
    for(x=0;c>='0'&&c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

struct edge{int to,next;}e[N<<1];
int n,cnt,head[N],clo[N],ans[N<<1];

inline void add(int x,int y) {
    e[++cnt].next=head[x],e[cnt].to=y,head[x]=cnt;
    e[++cnt].next=head[y],e[cnt].to=x,head[y]=cnt;
}

#define ans_push(x) ans[++ans[0]]=x

void dfs(int x,int fa) {
    clo[x]^=1;
    ans_push(x);
    for(int i=head[x];i;i=e[i].next)
        if(e[i].to!=fa) {
            dfs(e[i].to,x);
            ans_push(x);
            clo[x]^=1;
        }
    if(clo[x]==0)
        if(x==1) {
            ans_push(e[head[x]].to);
            ans_push(x);
            ans_push(e[head[x]].to);
        }
        else {
            ans_push(fa);
            ans_push(x);
            clo[fa]^=1;
            clo[x]^=1;
        }
}

int main() {
    read(n);
    int flag=0;
    for(int i=1;i<=n;++i) {
        read(clo[i]);
        if(clo[i]==-1) flag=1,clo[i]=0;
    }
    for(int i=1,x,y;i<n;++i) {
        read(x),read(y);
        add(x,y);
    }
    if(!flag) puts("1");
    else {
        clo[1]^=1;
        dfs(1,0);
        for(int i=1;i<=ans[0];++i)
            printf("%d%s",ans[i],i==ans[0]?"\n":" ");
    }
}