图论+DFS——Bubble Cup 9 - Finals [Online Mirror] E

• 题目链接：
  http://codeforces.com/contest/717/problem/E

• 参考链接：
  http://codeforces.com/contest/717/submission/20538605

• 分析：
  给出n个节点，和n个节点的颜色 1：黑色， -1： 粉红。再给出n-1条边，从1出发，每经过一个节点，节点的颜色将会变成另外一种。求一条路径，将所有的节点都变成黑色。

• 题解：
  可以直接从根节点开始DFS，每条分支都走一遍，直到都走成黑色再停止

• 参考代码：

/*************************************************************************
    > File Name: test.cpp
    > Author: Akira 
    > Mail: qaq.febr2.qaq@gmail.com 
 ************************************************************************/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#include <set>
#include <list>
#include <ctime>
typedef long long LL;
typedef unsigned long long ULL;
typedef long double LD;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Sqr(a) ((a)*(a))
using namespace std;

#define MaxN 200005
#define MaxM MaxN*2
#define INF 0x3f3f3f3f
#define bug cout<<88888888<<endl;

int n;
vector <int> vec[MaxN];

int color[MaxN];

void DFS(int x=0, int p=-1)
{
    printf("%d ", x+1);
    color[x] *= -1;
    for(int i=0;i<vec[x].size();i++)
    {
        if( vec[x][i] == p) continue;

        DFS(vec[x][i], x);

        if(x==0 && i == vec[x].size()-1 && color[vec[x][i]] == 1 && color[x] == 1) 
            return;

        printf("%d ", x+1);
        color[x] *= -1;

        if(color[vec[x][i]] == -1)
        {
            printf("%d ", vec[x][i]+1 );
            color[vec[x][i]] *= -1;

            if(x == 0 && i == vec[x].size()-1 && color[vec[x][i]] == 1 && color[x] == 1)
                return;

            printf("%d ", x+1);
            color[x] *= -1;
        }
    }
}

int main()
{
    while(~scanf("%d", &n))
    {   
        for(int i=0;i<n;i++)
        {
            cin >> color[i];
        }
        color[0]*=-1;
        for(int i=1;i<n;i++)
        {
            int u, v;
            scanf("%d%d", &u, &v);
            u--,v--;
            vec[u].push_back(v);
            vec[v].push_back(u);
        }
        DFS();
    }
    //system("pause");
}