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 =  + ;

 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 = ;
         while(x) ans += C[x], x -= lowbit(x);
         return ans;
     }

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

 }T[];

 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 + );
     post[u] = dfs_clock;
 }

 vector<PII> queries[maxn];

 bool ans[maxn];

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

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

     dfs(, );

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

     for(int i = ; 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], );

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

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

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

     return ;
 }

代码君