AtCoder ABC202E Count Descendants

题目链接:AtCoder ABC202E Count Descendants

题目大意:
给定一棵树,每次询问给出一个点\(u\)和深度\(d\),询问深度为\(d\)的点中有多少个点祖先包含\(u\)。

题解:
用\(dfs\)序,记录每一个点进入的时间\(in_i\)和出来的时间\(out_i\),则对于一个结点\(u\),其子树中某一结点\(v\)满足\(in_u < in_v < out_v < out_u\)。
由此,用\(vector\)维护相同深度下所有点的\(in\)(\(dfs\)时间戳是有序的,所以甚至不需要排序),然后每次二分查找\(in_u\)和\(out_u\)的位置并相减即可。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

struct Edge {
    int v, next;
} edge[400010];
int cnt, head[200010];
int n, in[200010], out[200010], tot, depth[200010];
int t;

void addEdge(int u, int v) {
    edge[++cnt].v = v;
    edge[cnt].next = head[u];
    head[u] = cnt;
}

vector<int> vec[200010];

void dfs(int u) {
    in[u] = ++tot;
    vec[depth[u]].push_back(in[u]);
    for (int i = head[u]; i; i = edge[i].next) {
        int v = edge[i].v;
        depth[v] = depth[u] + 1;
        dfs(v);
    }
    out[u] = ++tot;
}

int main() {
    ios::sync_with_stdio(false);
    cin >> n;
    for (int i = 2, x; i <= n; ++i) {
        cin >> x;
        addEdge(x, i);
    }
    dfs(1);
    cin >> t;
    while (t--) {
        int u, d;
        cin >> u >> d;
        cout << lower_bound(vec[d].begin(), vec[d].end(), out[u]) - lower_bound(vec[d].begin(), vec[d].end(), in[u]) << endl;
    }
    return 0;
}
上一篇:51Nod 2602 树的直径


下一篇:python编程语言基础