Cod sursa(job #2723915)

Utilizator levladiatorDragutoiu Vlad-Ioan levladiator Data 15 martie 2021 20:22:59
Problema Lowest Common Ancestor Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.73 kb
#include <bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define pf push_front
#define ll long long
#define ull unsigned long long
#define x first
#define y second
#define pi pair<int,int>
#define pl pair<ll,ll>

using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");

const ll N=1e5+5,INF=1e18,MOD=666013,M=1e2+5,inf=INT_MAX;

int n,m,k;
int euler[2*N],nivel[2*N],first[N],rmq[2*N][18];
vector<int> adj[250005];

void dfs(int nod,int lvl)
{
    euler[++k]=nod;
    nivel[k]=lvl;
    first[nod]=k;
    for(int i=0;i<adj[nod].size();i++)
    {
        int vec=adj[nod][i];
        dfs(vec,lvl+1);
        euler[++k]=nod;
        nivel[k]=lvl;
    }
}
void RMQ()
{
    for(int i=1;i<=k;i++)
    {
        rmq[i][0]=i;
    }
    for(int exp=1;(1<<exp)<=k;exp++)
    {
        for(int i=1;i<=k-(1<<exp)+1;i++)
        {
            int len=(1<<(exp-1));
            if(nivel[rmq[i][exp-1]]<nivel[rmq[i+len][exp-1]])
            {
                rmq[i][exp]=rmq[i][exp-1];
            }
            else rmq[i][exp]=rmq[i+len][exp-1];
        }
    }
}
int lca(int a,int b)
{
    int len=abs(first[a]-first[b])+1,ind=min(first[a],first[b]);
    int ans=ind;
    for(int bit=18;bit>=0;bit--)
    {
        if((1<<bit)&len)
        {
            if(nivel[rmq[ind][bit]]<nivel[ans])
            {
                ans=rmq[ind][bit];
            }
            ind+=(1<<bit);
        }
    }
    return euler[ans];
}

int main()
{
    fin>>n>>m;
    for(int i=2;i<=n;i++)
    {
        int x;
        fin>>x;
        adj[x].pb(i);
    }
    dfs(1,0);
    RMQ();
    for(int i=1;i<=m;i++)
    {
        int a,b;
        fin>>a>>b;
        fout<<lca(a,b)<<'\n';
    }

}