[ABC202E] Count Descendants

[ABC202E] Count Descendants

题意

​ 给出一棵以1为根的树，q次查询，每次查询输入$a_i,b_i$，问满足以下条件的u的个数：

• $a_i$位于u到1的最短路径上
• u到根的路径上恰好有$b_i$条边

分析

​ 根据需要满足的条件可知，路径上恰好有$b_i$条边，也就是说明u的深度为$b_i$，那么就可以存下来每一层都有哪些节点，然后再考虑如何判断$a_i$是否位于u到1的最短路径上，这里考虑dfs序，根据dfs序可知，如果$a_i$位于u到1的最短路径上，说明$a_i$比u遍历的更早，且出去的更晚，这里可以存下来dfs序，然后在深度上进行二分，找到上边界和下边界，二者相减即为答案

AC代码

#include <bits/stdc++.h>
using namespace std;
using LL = long long;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n;
    cin >> n;
    vector<vector<int> > G(n + 1);
    for (int i = 2; i <= n; i++) {
        int u;
        cin >> u;
        G[u].push_back(i);
        G[i].push_back(u);
    }
    int cnt = 0;
    vector<vector<int> > G1(n + 1);
    vector<int> dep(n + 1), in(n + 1), out(n + 1);
    function<void(int, int)> dfs = [&](int u, int fa) {
        in[u] = ++cnt;
        if (u != 1) {
            dep[u] = dep[fa] + 1;
        }
        G1[dep[u]].push_back(in[u]);
        for (auto v : G[u]) {
            if (v != fa) {
                dfs(v, u);
            }
        }
        out[u] = ++cnt;
    };
    dfs(1, 0);
    int q;
    cin >> q;
    while (q--) {
        int x, y;
        cin >> x >> y;
        int ans = upper_bound(G1[y].begin(), G1[y].end(), out[x]) - G1[y].begin();
        ans = ans - (lower_bound(G1[y].begin(), G1[y].end(), in[x]) - G1[y].begin());
        cout << max(ans, 0) << '\n';
    }
    return 0;
}