Codeforces Round #316 (Div. 2) D. Tree Requests

用深搜为每个结点建立一个时间序in[], out[]，在搜索过程中把在同一深度，同一颜色结点的时间保存起来。查询的时候，若结点为v, 则所查询的所有节点的时间序在in[v]和out[v]之间，再用二分查找各个颜色结点个数。

#include <bits/stdc++.h>
#define maxn 500005
using namespace std;

vector<int> v[maxn];
vector<int> f[maxn][26];
int n, m;
string s;
int in[maxn], out[maxn], tot;
void dfs(int h, int x){

in[x] = ++tot;
f[h][s[x-1]-'a'].push_back(tot);

for(int i = 0; i < v[x].size(); i++)
dfs(h+1, v[x][i]);

out[x] = tot;
}
int main(){

//  freopen("in.txt", "r", stdin);
int a;
scanf("%d%d", &n, &m);
for(int i = 2; i <= n; i++){
scanf("%d", &a);
v[a].push_back(i);
}
cin >> s;
tot = 0;
dfs(1, 1);

int v, h;
for(int i = 0; i < m; i++){
scanf("%d%d", &v, &h);
int p = 0;
for(int j = 0; j < 26; j++){
int d = upper_bound(f[h][j].begin(), f[h][j].end(), out[v]) -
lower_bound(f[h][j].begin(), f[h][j].end(), in[v]);
if(d&1)
p++;
if(p > 1)
break;
}
if(p > 1)
puts("No");
else
puts("Yes");
}

return 0;
}