CodeForces 570D DFS序 树状数组 Tree Requests

参考九野巨巨的博客。

查询一个子树内的信息，可以通过DFS序转成线形的，从而用数据结构来维护。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#define MP make_pair
#define FI first
#define SE second
using namespace std;

typedef pair<int, int> PII;

const int maxn = 500000 + 10;

int n, m;

vector<int> G[maxn];
char s[maxn];

struct BIT
{
    int C[maxn];

    inline int lowbit(int x) { return x & (-x); }

    void add(int x, int v)
    {
        while(x <= n) C[x] += v, x += lowbit(x);
    }

    int sum(int x)
    {
        int ans = 0;
        while(x) ans += C[x], x -= lowbit(x);
        return ans;
    }

    int Query(int l, int r) { return sum(r) - sum(l - 1); }

}T[26];

vector<int> Dep[maxn];
int dfs_clock;
int pre[maxn], post[maxn];

void dfs(int u, int dep)
{
    Dep[dep].push_back(u);
    pre[u] = ++dfs_clock;
    for(int v : G[u]) dfs(v, dep + 1);
    post[u] = dfs_clock;
}

vector<PII> queries[maxn];

bool ans[maxn];

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 2; i <= n; i++)
    {
        int u; scanf("%d", &u);
        G[u].push_back(i);
    }
    scanf("%s", s + 1);

    int maxh = 1;
    for(int i = 0; i < m; i++)
    {
        int v, h; scanf("%d%d", &v, &h);
        maxh = max(maxh, h);
        queries[h].push_back(MP(v, i));
    }

    dfs(1, 1);

    for(int i = 0; i < m; i++) ans[i] = true;

    for(int i = 1; i <= maxh; i++)
    {
        if(Dep[i].empty()) break;
        if(queries[i].empty()) continue;

        for(int v : Dep[i]) T[s[v] - 'a'].add(pre[v], 1);

        for(PII x : queries[i])
        {
            int odd = 0;
            for(int j = 0; j < 26; j++)
            {
                odd += (T[j].Query(pre[x.FI], post[x.FI]) & 1);
            }
            if(odd >= 2) ans[x.SE] = false;
        }

        for(int v : Dep[i]) T[s[v] - 'a'].add(pre[v], -1);
    }

    for(int i = 0; i < m; i++) printf("%s\n", ans[i] ? "Yes" : "No");

    return 0;
}