Cod sursa(job #3004437)

Utilizator T1raduTaerel Radu Nicolae T1radu Data 16 martie 2023 12:27:25
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
int n,t,tt[250001][20],p[20];
int main()
{
    fin >> n >> t;
    for(int i=1;i<=n;i++)
    {
        int x;
        fin >> x;
        tt[i][0]=x;
    }
    int px=0;
    p[0]=1;
    while(p[px]<=n)
    {
        px++;
        p[px]=p[px-1]*2;
    }
    for(int q=1;q<=px;q++)
    {
        for(int i=1;i<=n;i++)
            tt[i][q]=tt[tt[i][q-1]][q-1];
    }
    for(int i=1;i<=t;i++)
    {
        int nod,q;
        fin >> nod >> q;
        while(q>0)
        {
            int poz=lower_bound(p,p+px+1,q) - p;
            if(q<p[poz]) poz--;
            nod=tt[nod][poz];
            q-=p[poz];
        }
        fout << nod << "\n";
    }
    return 0;
}