P6135 【模板】虚树

一个纯虚树的模板吧，不涉及啥要处理的东西。

题目描述
给定一棵 n 个点的有根树，树上有 k 个关键点，请你构建这些点的虚树。

输入格式
第一行两个整数 n,k。

第二行 n 个整数 f 1…n，其中 fi 表示 i 的父亲。特别地，若 i 为根，则 fi =0。

第三行 k 个整数，表示关键点。

输出格式
共 n 行，第 i 行两个整数 gi, di

若 i 不在虚树中，则 gi = di = -1

若 i 为虚树的根，则 gi = di = 0

若 i 在虚树中但不为根，则 gi 为 i 在虚树中的父亲，di表示 i 和 gi在原树中的距离。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<deque>
#include<map>
#include<vector>
#include<cmath>
#define ll long long
#define llu unsigned ll
using namespace std;
const double eps = 1e-8;
const ll lnf = 0x3f3f3f3f3f3f3f3f;
const int inf = 0x3f3f3f3f;
const int maxn = 200100;
const int mod = 1e9 + 7;
int head[maxn], ver[maxn], nt[maxn];
int fa[maxn], d[maxn], f[maxn][22], dfn[maxn], a[maxn],s[maxn];
int t, tot = 1, cnt = 0, top = 0;
int n, k, x, rt;


void add(int x, int y)
{
    ver[++tot] = y, nt[tot] = head[x], head[x] = tot;
}


bool cmp(const int& a, const int& b)
{
    return dfn[a] < dfn[b];
}


void dfs(int x)
{
    dfn[x] = ++cnt;
    for (int i = head[x];i;i = nt[i])
    {
        int y = ver[i];
        d[y] = d[x] + 1;
        f[y][0] = x;
        for (int j = 1;j <= t;j++)
            f[y][j] = f[f[y][j - 1]][j - 1];
        dfs(y);
    }
}

int LCA(int x, int y)
{
    if (d[x] > d[y]) swap(x, y);
    for (int i = t;i >= 0;i--)
        if (d[f[y][i]] >= d[x]) y = f[y][i];
    if (y == x) return x;
    for (int i = t;i >= 0;i--)
        if (f[x][i] != f[y][i]) x = f[x][i], y = f[y][i];
    return f[x][0];
}

void in(int x)
{
    if (x == 0) return;
    if (top == 1)
    {
        s[++top] = x;
        return;
    }
    int lca = LCA(s[top], x);
    if (lca == s[top])
    {
        s[++top] = x;
        return;
    }
    while (top > 1 && dfn[s[top - 1]] >= dfn[lca])
    {
        fa[s[top]] = s[top - 1];
        top--;
    }
    if (lca != s[top]) fa[s[top]]=lca, s[top] = lca;
    s[++top] = x;

}

void build(int k)
{
    for (int i = 1;i <= k;i++)
        scanf("%d", &a[i]);
    sort(a + 1, a + k + 1, cmp);

    s[top = 1] = 0;
  
    for (int i = 1;i <= k;i++)
        in(a[i]);
    while (top) fa[s[top]]=s[top-1],top--;
}


int main(void)
{

    scanf("%d%d", &n,&k);
    t = log(n) / log(2) + 1;

    for (int i = 1;i <= n;i++)
    {
        scanf("%d", &x);
        if (x == 0) rt = i;
        add(x, i);
    }

    memset(fa, -1, sizeof(fa));

    d[rt] = 1;
    dfs(rt);

    build(k);
    

    for (int i = 1;i <= n;i++)
    {
        if (fa[i] == -1) printf("-1 -1\n");
        else if (fa[i] == 0) printf("0 0\n");
        else printf("%d %d\n", fa[i], d[i] - d[fa[i]]);
    }

    return 0;

}