标签：Count AtCoder ABC202E int cnt dfs 200010 edge out

题目链接：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;
}

标签：Count,AtCoder,ABC202E,int,cnt,dfs,200010,edge,out
来源： https://www.cnblogs.com/IzumiSagiri/p/15860421.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。