【BZOJ 5165】 树上倍增

【题目链接】

点击打开链接

【算法】

树上倍增，时间复杂度 ： O(qklog(n))

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 3000010
#define MAXLOG 18
const int INF = 1e8;

int T,tot = ,i,x;
char opt[];
int q[MAXN],dep[MAXN],anc[MAXN][MAXLOG];

inline void update(int x)
{
    int i;
    tot++;
    dep[tot] = dep[x] + ;
    anc[tot][] = x;
    for (i = ; i < MAXLOG; i++) anc[tot][i] = anc[anc[tot][i-]][i-];
}
inline int query(int k)
{
    int i,j,t,mn = INF;
    bool flag = true;
    for (i = ; i <= k; i++) mn = min(mn,dep[q[i]]);
    for (i = ; i <= k; i++)
    {
        t = dep[q[i]] - mn;
        for (j = ; j < MAXLOG; j++)
        {
            if (t & ( << j))
                q[i] = anc[q[i]][j];
        }
    }
    for (i = ; i <= k; i++) flag &= (q[i] == q[]);
    if (flag) return q[];
    for (i = MAXLOG - ; i >= ; i--)
    {
        flag = true;
        for (j = ; j <= k; j++) flag &= (anc[q[j]][i] == anc[q[]][i]);
        if (!flag)
        {
            for (j = ; j <= k; j++) q[j] = anc[q[j]][i];
        }
    }
    return anc[q[]][];
}

int main()
{

    scanf("%d",&T);
    while (T--)
    {
        scanf("%s%d",&opt,&x);
        if (opt[] == 'A') update(x);
        else
        {
            for (i = ; i <= x; i++) scanf("%d",&q[i]);
            printf("%d\n",query(x));
        }
    }
    return ;
}