Cod sursa(job #2682608)

Utilizator Iulia_DianaIulia Diana Iulia_Diana Data 8 decembrie 2020 23:54:32
Problema Lowest Common Ancestor Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.56 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
int w[100005], dp[20][100005], euler[200005], nr, poz[100005], h[100005];
vector <int> v[100005];
void dfs(int nod)
{
    nr++;
    euler[nr]=nod;
    poz[nod]=nr;
    for(int i=0; i<v[nod].size(); i++)
        if(h[v[nod][i]]==0)
        {
            h[v[nod][i]]=h[nod]+1;
            dfs(v[nod][i]);
            nr++;
            euler[nr]=nod;
        }


}
int main()
{
    int n, m;
    fin >> n >> m;
    for(int i=2; i<=n; i++)
    {
        int dad;
        fin >> dad;
        v[dad].push_back(i);
    }
    h[1]=1;
    dfs(1);
//    for(int i=1; i<=nr; i++)   fout << euler[i] << " ";
//    fout << "\n";
//    for(int i=1; i<=n; i++)   fout << poz[i] << " ";
//    fout << "\n";
//    for(int i=1; i<=n; i++)   fout << h[i] << " ";
//    fout << "\n"; fout << "\n"; fout << "\n";

    for(int i=1; i<=nr; i++)
        dp[0][i]=euler[i];
    for(int i=1; (1<<i)<=nr; i++)
    {
        for(int j=1; j<=nr; j++)
        {
            if(h[dp[i-1][j]]<h[dp[i-1][j+(1<<(i-1))]] || j+(1<<(i-1))>nr)  dp[i][j]=dp[i-1][j];
            else   dp[i][j]=dp[i-1][j+(1<<(i-1))];
         //   fout << dp[i][j] << " ";
        }
      //  fout << "\n";
    }
    for(int i=1; i<=m; i++)
    {
        int a, b;
        fin >> a >> b;
        int j;
        for(j=1; (1<<j)<=poz[b]-poz[a]+1; j++);
        j--;
        fout << min(dp[j][poz[a]], dp[j][poz[b]-(1<<j)+1]) << "\n";
    }

    return 0;
}