Cod sursa(job #2922344)

Utilizator NutaAlexandruASN49K NutaAlexandru Data 7 septembrie 2022 23:09:53
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.13 kb
#include <fstream>
#include <vector>
#include <cmath>
#include <algorithm>
#include <set>
#include <queue>
using namespace std;
class lca
{
    vector<vector<int>>g,rmq;
    vector<int>e,poz,h,lg;
    int n;
public:
    lca(vector<vector<int>>a)
    {
        g=a;
        n=(int)a.size()-1;
        poz.resize(n+1);
        h.resize(n+1);
        lg.resize(2*n+1);
        lg[0]=lg[1]=0;
        for(int i=2;i<=2*n;i++)
        {
            lg[i]=lg[i/2]+1;
        }
        h[1]=0;
        e={0};
        rmq=vector<vector<int>>(2*n+1,vector<int>(lg[2*n]+2));
    }
    void dfs(int nod,int tata)
    {
        h[nod]=h[tata]+1;
        poz[nod]=(int)e.size();
        e.push_back(nod);
        for(auto c:g[nod])
        {
            if(c==tata)continue;
            dfs(c,nod);
            e.push_back(nod);
        }
    }
    void dfs()
    {
        dfs(1,1);
    }
    int maxx(int x,int y)
    {
        if(h[x]>h[y])return y;
        return x;

    }
    void build_rmq()
    {
        n=(int)e.size()-1;
        for(int i=1;i<=n;i++)
        {
            rmq[i][0]=e[i];
        }
        for(int j=1;(1<<j)<=n;j++)
        {
            for(int i=1;i+(1<<j)-1<=n;i++)
            {
                rmq[i][j]=maxx(rmq[i][j-1],rmq[i+(1<<(j-1))][j-1]);
            }
        }
    }
    void pre()
    {
        dfs();
        build_rmq();
    }
    int get_lca(int x, int y)
    {
        int l=poz[x],r=poz[y];
        if(r<l)
        {
            swap(l,r);
        }
        int lung=lg[r-l+1];
        return maxx(rmq[l][lung],rmq[r-(1<<lung)+1][lung]);
    }

};

int main()
{
    ifstream cin("lca.in");
    ofstream cout("lca.out");
    int n,q;
    cin>>n>>q;
    vector<int>val(n+1,0);
    for(int i=1;i<=n;i++)
    {
        //cin>>val[i];
    }
    vector<vector<int>>a(n+1);
    for(int i=2;i<=n;i++)
    {
        int x,y;
        //cin>>x>>y;
        cin>>x;
        a[x].push_back(i);
        //a[y].push_back(x);
    }
    lca tree(a);
    tree.pre();
    int x,y;
    while(cin>>x>>y)
    {
        cout<<tree.get_lca(x,y)<<'\n';
    }
}