题目:
http://codeforces.com/contest/570/problem/D
题意:
n个点的一棵树,每个节点都有自己的字母,m个询问,问u节点的子树中高度为h的点是否可以组成回文序列。
思路:
dfs,求出每个点的dfs序,以及记录每个深度的节点的字母的序。
字母数量的奇数个不超过1,则可以组成回文序列。二分求出询问节点对应的区间,判断是否可以。
AC.
#include
#include
#include
#include
#include
using namespace std;
const int maxn = 5e5+5;
int n, m;
char let[maxn];
int tol, head[maxn];
struct Edge{
int to, next;
}edge[maxn];
void addedge(int u, int v)
{
edge[tol].to = v;
edge[tol].next = head[u];
head[u] = tol++;
}
vector vec[maxn][30];
int in[maxn], ou[maxn], tot;
void dfs(int u, int d)
{
in[u] = ++tot;
vec[d][let[u]-'a'].push_back(tot);
for(int i = head[u]; ~i; i = edge[i].next) {
int v = edge[i].to;
dfs(v, d+1);
}
ou[u] = ++tot;
}
void init()
{
tol = 0;
tot = 0;
memset(head, -1, sizeof(head));
for(int i = 0; i <= n; ++i) {
for(int j = 0; j <= 26; ++j) {
vec[i][j].clear();
}
}
}
int main()
{
//freopen("in", "r", stdin);
while(~scanf("%d%d", &n, &m)) {
int u, v;
init();
for(int i = 2; i <= n; ++i) {
scanf("%d", &u);
addedge(u, i);
}
scanf("%s", let+1);
dfs(1, 1);
while(m--) {
scanf("%d %d", &u, &v);
int res = 0, ok = 0;
for(int i = 0; i < 26; ++i) {
res = lower_bound(vec[v][i].begin(), vec[v][i].end(), ou[u]) -
lower_bound(vec[v][i].begin(), vec[v][i].end(), in[u]-1);
if(res % 2) ok++;
if(ok > 1) break;
}
if(ok > 1) printf("No\n");
else printf("Yes\n");
}
}
return 0;
}