题目链接:http://codeforces.com/contest/717/problem/E
题意:给一棵树 有两个颜色,一个人从节点1出发,每走到一个节点都会使对应节点变色。问怎么走可以让整棵树变成同一个颜色。
很容易想到分子树去处理,假如一棵深度为2的子树,根节点有n个儿子。儿子里有k个是反色的,那么反色的点必然会被走一次,同时根节点也会跟着变色。当这棵子树搞定以后,这个节点就会变成上一棵子树的叶节点,同样的操作即可。所以关注两层节点,后序遍历后如果儿子未变成1,那么和父亲来回走一遍。注意根节点1,整棵树走完以后有可能根节点未变成1,只要再随便找一个点走一遍,最终落脚到那个点就行了。
1 /* 2 ━━━━━┒ギリギリ♂ eye! 3 ┓┏┓┏┓┃キリキリ♂ mind! 4 ┛┗┛┗┛┃\○/ 5 ┓┏┓┏┓┃ / 6 ┛┗┛┗┛┃ノ) 7 ┓┏┓┏┓┃ 8 ┛┗┛┗┛┃ 9 ┓┏┓┏┓┃ 10 ┛┗┛┗┛┃ 11 ┓┏┓┏┓┃ 12 ┛┗┛┗┛┃ 13 ┓┏┓┏┓┃ 14 ┃┃┃┃┃┃ 15 ┻┻┻┻┻┻ 16 */ 17 #include <algorithm> 18 #include <iostream> 19 #include <iomanip> 20 #include <cstring> 21 #include <climits> 22 #include <complex> 23 #include <fstream> 24 #include <cassert> 25 #include <cstdio> 26 #include <bitset> 27 #include <vector> 28 #include <deque> 29 #include <queue> 30 #include <stack> 31 #include <ctime> 32 #include <set> 33 #include <map> 34 #include <cmath> 35 using namespace std; 36 #define fr first 37 #define sc second 38 #define cl clear 39 #define BUG puts("here!!!") 40 #define W(a) while(a--) 41 #define pb(a) push_back(a) 42 #define Rint(a) scanf("%d", &a) 43 #define Rll(a) scanf("%I64d", &a) 44 #define Rs(a) scanf("%s", a) 45 #define Cin(a) cin >> a 46 #define FRead() freopen("in", "r", stdin) 47 #define FWrite() freopen("out", "w", stdout) 48 #define Rep(i, len) for(int i = 0; i < (len); i++) 49 #define For(i, a, len) for(int i = (a); i < (len); i++) 50 #define Cls(a) memset((a), 0, sizeof(a)) 51 #define Clr(a, x) memset((a), (x), sizeof(a)) 52 #define Full(a) memset((a), 0x7f7f7f, sizeof(a)) 53 #define lrt rt << 1 54 #define rrt rt << 1 | 1 55 #define pi 3.14159265359 56 #define RT return 57 #define lowbit(x) x & (-x) 58 #define onecnt(x) __builtin_popcount(x) 59 typedef long long LL; 60 typedef long double LD; 61 typedef unsigned long long ULL; 62 typedef pair<int, int> pii; 63 typedef pair<string, int> psi; 64 typedef pair<LL, LL> pll; 65 typedef map<string, int> msi; 66 typedef vector<int> vi; 67 typedef vector<LL> vl; 68 typedef vector<vl> vvl; 69 typedef vector<bool> vb; 70 71 const int maxn = 200200; 72 int n; 73 int color[maxn]; 74 vector<int> G[maxn]; 75 76 void dfs(int u, int p) { 77 Rep(i, G[u].size()) { 78 int v = G[u][i]; 79 if(v == p) continue; 80 color[v] = -color[v]; 81 printf("%d ", v); 82 dfs(v, u); 83 color[u] = -color[u]; 84 printf("%d ", u); 85 if(color[v] == -1) { 86 color[v] = -color[v]; 87 printf("%d ", v); 88 color[u] = -color[u]; 89 printf("%d ", u); 90 } 91 } 92 } 93 94 int main() { 95 // FRead(); 96 int u, v; 97 while(~Rint(n)) { 98 Cls(color); 99 For(i, 1, n+1) { 100 Rint(color[i]); 101 G[i].clear(); 102 } 103 Rep(i, n-1) { 104 Rint(u); Rint(v); 105 G[u].push_back(v); 106 G[v].push_back(u); 107 } 108 printf("1 "); 109 dfs(1, 0); 110 if(color[1] == -1) { 111 int v = G[1][0]; 112 color[v] = -color[v]; 113 printf("%d ", v); 114 color[1] = -color[1]; 115 printf("%d ", 1); 116 color[v] = -color[v]; 117 printf("%d ", v); 118 } 119 } 120 RT 0; 121 }