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

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;
}