CodeForces 208E Blood Cousins

最新推荐文章于 2022-03-12 21:17:01 发布

X_rotator

最新推荐文章于 2022-03-12 21:17:01 发布

阅读量388

点赞数

分类专栏：题解文章标签： codeforces 树

本文链接：https://blog.csdn.net/X_rotator/article/details/64126438

版权

题解专栏收录该内容

19 篇文章 0 订阅

订阅专栏

CodeForces 208E Blood Cousins

题目描述：

给一片有根森林（我知道不是这么个说法但自己意会就可以了），进行若干次询问，每次询问 $v, p$ ，输出除了点 $v$ 之外还有多少点的 $p$ 阶祖先和点 $v$ 的一样。

题解：

首先 $p$ 阶祖先这个东西可以倍增搞出来。

对于每一组询问，找到 $v$ 的 $p$ 阶祖先 $u$ ，那么实际上就是求 $u$ 的子树当中有多少点的深度和点 $v$ 相同。我们可以对于每一个不同的深度，记录节点的 DFS 序，这样询问的时候在对应的深度里二分就行了。

题目链接： vjudge 原网站

代码：


#include <vector>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
#include <algorithm>
using namespace std;

#define MAXN 100010

static struct
{
    int v, next;
} edge[MAXN];
static int N, M, G, dfn_tot;
static int fa[MAXN][20], head[MAXN], dep[MAXN], dfn[MAXN][2];
static vector<int> pos[MAXN];

inline void add_edge(int u, int v)
{
    edge[G].v = v; edge[G].next = head[u]; head[u] = G++;
}

void dfs(int x)
{
    for (int i = 1; i < 20; i++)
        fa[x][i] = fa[fa[x][i-1]][i-1];
    pos[dep[x] = dep[fa[x][0]] + 1].push_back(dfn[x][0] = ++dfn_tot);
    for (int i = head[x]; ~i; i = edge[i].next)
        dfs(edge[i].v);
    dfn[x][1] = dfn_tot;
}

int main()
{
    memset(head, -1, sizeof(head));
    scanf("%d", &N);
    for (int i = 1; i <= N; i++)
    {
        scanf("%d", &fa[i][0]);
        if (fa[i][0]) add_edge(fa[i][0], i);
    }
    for (int i = 1; i <= N; i++)
        if (!fa[i][0]) dfs(i);
    scanf("%d", &M);
    while (M--)
    {
        int v, p, l, r, lca;
        scanf("%d%d", &v, &p);
        lca = v;
        for (int i = 0; i < 20; i++)
            if (p & (1 << i)) lca = fa[lca][i];
        if (!lca)
            printf("0 ");
        else
        {
            l = lower_bound(pos[dep[v]].begin(), pos[dep[v]].end(), dfn[lca][0] + 1) - pos[dep[v]].begin();
            r = upper_bound(pos[dep[v]].begin(), pos[dep[v]].end(), dfn[lca][1]) - pos[dep[v]].begin();
            printf("%d ", r - l - 1);
        }
    }
    return 0;
}