D. Tree Requests
题目大意
一棵树上的包含N个节点 (N <= 500000) 每个节点包含一个小写字母,每次询问包含u, h,询问u节点所在的子树中所有深度为h的节点是否等够组成一个回文串(顺序可以随意排列)。
题目分析
能够组成回文串的充要条件是最多有一个字母其出现次数为奇数。
关键是如何统计字母的出现次数。
首先记录每个的节点DFS的首次经过的时间in[]和结束经过的时间out[]。我们将所有的节点重新编号为DFS首次经过的时间。
那么所有u的子树的节点可以表示为区间[in[u], out[u]]。
我们还需要记录每层的节点有哪些。于是我们就可以用二分查找h深度的所有在u的子树的区间。
为了节约空间和加快运算速度,可以用位运算,用26个位表示26个字母奇偶性,用异或代替+-运算。
参考代码
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int INF = 0x7fffffff;
const int MOD = 1e9 + 7;
const int N = 5e5 + 10;
vector <int> H[N];
vector <int> mask[N];
vector <int> G[N];
int in[N], out[N];
char str[N];
int cnt;
void dfs(int u, int depth = 1) {
in[u] = ++cnt;
H[depth].push_back(cnt);
mask[depth].push_back(mask[depth].back() ^ (1 << (str[u] - 'a')));
for(int i = 0; i < G[u].size(); i++)
dfs(G[u][i], depth + 1);
out[u] = cnt;
}
int main() {
#ifdef Tally_Ho
freopen("in.txt", "r", stdin);
#endif // Tally_Ho
int n, m, p;
scanf("%d%d", &n, &m);
for(int i = 1; i < n; i++) {
scanf("%d", &p);
p--;
G[p].push_back(i);
}
for(int i = 0;i<=n;i++) {
H[i].push_back(0);
mask[i].push_back(0);
}
scanf("%s", str);
cnt = 0;
dfs(0);
int v, h;
for(int i = 0; i < m; i++) {
scanf("%d %d", &v, &h);
v--;
int L = lower_bound(H[h].begin(), H[h].end(), in[v]) - H[h].begin();
int R = upper_bound(H[h].begin(), H[h].end(), out[v]) - 1 - H[h].begin();
int x = mask[h][R] ^ mask[h][L - 1];
if(x - (x & -x) == 0)
puts("Yes");
else
puts("No");
}
return 0;
}