58 lines
1.4 KiB
C++
58 lines
1.4 KiB
C++
#include <bits/stdc++.h>
|
||
using namespace std;
|
||
const int N = 1e5 + 10;
|
||
int n;
|
||
int son[N][2]; // 记录每个节点的两个儿子
|
||
int fa[N]; // 记录每个节点的父节点是哪个
|
||
int col[N]; // 记录每个节点的颜色
|
||
int sum[N]; // 记录每个节点被操作的次数
|
||
|
||
void dfs(int x, int now) {
|
||
now += sum[x]; // 递归计算x节点被操作的次数
|
||
if (now & 1)
|
||
col[x] ^= 1; // 如果操作次数是奇数，则颜色取反
|
||
for (int i = 0; i < 2; i++) {
|
||
if (son[x][i] != -1)
|
||
dfs(son[x][i], now);
|
||
}
|
||
}
|
||
|
||
int main() {
|
||
#ifndef ONLINE_JUDGE
|
||
freopen("20240602.in", "r", stdin);
|
||
#endif
|
||
|
||
cin >> n; // n个节点
|
||
memset(son, -1, sizeof son); // 初始化son数组,-1表示没有儿子
|
||
// 这个i从2开始，是因为题目中要求的意义
|
||
for (int i = 2; i <= n; i++) {
|
||
cin >> fa[i];
|
||
for (int j = 0; j < 2; j++) { // 循环两次，因为每个节点有两个儿子
|
||
if (son[fa[i]][j] == -1) {
|
||
son[fa[i]][j] = i; // 记录父子关系
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
string s;
|
||
cin >> s;
|
||
// 初始颜色
|
||
for (int i = 1; i <= n; i++)
|
||
col[i] = s[i - 1] - '0';
|
||
|
||
int q;
|
||
cin >> q;
|
||
while (q--) {
|
||
int x;
|
||
cin >> x;
|
||
sum[x] += 1; // x这个节点被操作了一次
|
||
}
|
||
// 从根节点开始dfs
|
||
dfs(1, 0);
|
||
// 输出每个节点的颜色
|
||
for (int i = 1; i <= n; i++)
|
||
cout << col[i];
|
||
|
||
return 0;
|
||
}
|