Cod sursa(job #2281807)

Utilizator PredescuSebastianIonPredescu Sebastian Ion PredescuSebastianIon Data 12 noiembrie 2018 19:23:38
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.66 kb
#include <fstream>

using namespace std;
ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
int a[20][250002],b[250002],n,m,q,p;
int solve(int q,int p)
{
    while(p)
    {
        int k=b[p];
        q=a[k][q];
        p=p-(1<<k);
    }
    return q;
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>a[0][i];
    }
    for(int i=2;i<=n;i++)
    {
        b[i]=b[i/2]+1;
    }
    for(int i=1;1<<i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            a[i][j]=a[i-1][a[i-1][j]];
        }
    }
    while(m--)
    {
        fin>>q>>p;
        fout<<solve(q,p)<<'\n';
    }
    return 0;
}