Codeforces Round #316 (Div. 2) D. Tree Requests 启发式合并

题意

一颗根在1的树，多次询问，每次询问$v$的位于深度$h$的子节点是否能构成回文串。

分析

最多有一奇数字符才能构成回文串。
考虑将询问离线，使用启发式合并求解该问题。
维护$cnt[h][ch]$数组，表示在深度为$h$的位置上字符$ch$的字符个数。

遍历所有轻儿子，递归结束消除贡献
遍历重儿子，保留重儿子贡献
计算当前子树中轻子树的贡献
更新答案
当前点是轻儿子，消除当前子数贡献

卡$endl$，用\n

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define endl '\n'
const int N = 1e6 + 10, M = N * 2;
int n, m;
int h[N], e[M], ne[M], idx;
int sz[N], d[N], son[N];
int cnt[N][26], ans[N];
char c[N];
vector<pii> G[N];
void add(int a, int b){
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
int dfs(int u, int fa, int dep){
    d[u] = dep, sz[u] = 1;
    for(int i = h[u]; ~i; i = ne[i]){
        int j = e[i];
        if(j == fa) continue;
        sz[u] = sz[u] + dfs(j, u, dep + 1);
        if(sz[son[u]] < sz[j]) son[u] = j;
    }
    return sz[u];
}
void update(int u, int fa, int sgn, int flag){
    cnt[d[u]][c[u]-'a'] ++ ;
    for(int i = h[u]; ~i; i = ne[i]){
        int j = e[i];
        if(j == fa || j == flag) continue;
        update(j, u, sgn, flag);
    }
}
void dfs1(int u, int fa, int op){
    for(int i = h[u]; ~i; i = ne[i]){
        int j = e[i];
        if(j == fa || j == son[u]) continue;
        dfs1(j, u, 0);
    }
    if(son[u]) dfs1(son[u], u, 1);
    update(u, fa, 1, son[u]);
    for(auto it : G[u]){
        int d = it.first, now = 0;
        for(int i = 0; i < 26; i++)
            if(cnt[d][i] % 2) now++;
        if(now <= 1) ans[it.second] = 1;
    }
    if(op == 0) update(u, fa, -1, 0);
}
int main(){
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    memset(h, -1, sizeof(h));
    cin>>n>>m;
    for(int i = 2; i <= n; i++){
        int x; cin>>x;
        add(x, i), add(i, x);
    }
    cin>>(c + 1);
    for(int i = 1; i <= m; i++){
        int x, h; cin>>x>>h;
        G[x].push_back({h, i});
    }
    dfs(1, -1, 1);
    dfs1(1, -1, 0);
    for(int i = 1; i <= m; i++) cout<<(ans[i] == 1 ? "Yes" : "No")<<endl;
    return 0;
}