Cod sursa(job #1772799)

Utilizator RaduMirceaAndreiRadu Mircea Andrei RaduMirceaAndrei Data 7 octombrie 2016 00:34:41
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.7 kb
# include <fstream>
# define DIM 250010
using namespace std;
ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
int d[20][DIM],lg[DIM],p2[20],n,m,p,q,i,j,step;
int main () {
    fin>>n>>m;
    p2[0]=1;
    for(i=1;p2[i-1]*2<=n;i++)
        p2[i]=p2[i-1]*2;
    for(i=2;i<=n;i++)
        lg[i]=lg[i/2]+1;
    for(i=1;i<=n;i++)
        fin>>d[0][i];
    for(i=1;i<=19;i++)
        for(j=1;j<=n;j++)
            d[i][j]=d[i-1][d[i-1][j]];
    for(i=1;i<=m;i++){
        fin>>q>>p;
        step=p2[lg[p]];
        for(j=0;step;step>>=1)
            if(j+step<=p){
                q=d[lg[step]][q];
                j+=step;
            }
        fout<<q<<"\n";
    }
    return 0;
}