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 100010static 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;}

提交记录（AC / Total = 1 / 1）：

Run ID Remote Run ID Time(ms) Memory(kb) Result Submit Time 8500822 25660729 248 15720 AC 2017-03-20 15:11:48