Cod sursa(job #2848341)

Utilizator rafaelrafyChitan Rafael rafaelrafy Data 12 februarie 2022 13:28:51
Problema Lowest Common Ancestor Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.43 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <climits>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
vector<int>v[100100];
vector<int>E; int k;
int ap[100100];
int nivel[100100];
int x,y,n,m;
int aint[600100];
void dfs(int x,int niv)
{
    nivel[x]=niv;
    E.push_back(x);
    k++;
    ap[x]=k;
    for(auto i:v[x])
        if(!ap[i])
        {
            dfs(i, niv+1);
            E.push_back(x);
            k++;
        }
}
void generare(int x,int y,int poz)
{
    if(x==y)
    {
        aint[poz]=E[x];
        return;
    }
    generare(x, (x+y)/2, 2*poz);
    generare((x+y)/2+1, y, 2*poz+1);
    
    if(nivel[aint[2*poz]] < nivel[aint[2*poz+1]])
        aint[poz] = aint[2*poz];
    
    else
        aint[poz] = aint[2*poz+1];
}
int query(int x,int y,int a,int b,int poz)
{
    if(y<a || x>b)
        return 0;
    
    if(x<=a && y>=b)
        return aint[poz];
    
    int q1=query(x, y, a, (a+b)/2, 2*poz), q2=query(x, y, (a+b)/2+1, b, 2*poz+1);
    
    if(nivel[q1] < nivel[q2])
        return q1;
    
    return q2;
}
int main() {
    nivel[0]=INT_MAX;
    E.push_back(0);
    fin>>n>>m;
    for(int i=2;i<=n;i++)
    {
        fin>>x;
        v[x].push_back(i);
        v[i].push_back(x);
    }
    dfs(1,1);
    generare(1, E.size()-1, 1);
    
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y;
        fout<<query(min(ap[x],ap[y]),max(ap[x],ap[y]),1,E.size()-1,1)<<'\n';
    }
    return 0;
}