Cod sursa(job #1647788)

Utilizator bt.panteaPantea Beniamin bt.pantea Data 10 martie 2016 22:06:31
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f ("stramosi.in");
ofstream g ("stramosi.out");
int r[300000];
int log2(int p)
{
    int sol=0;
    while (p>1)
        sol++,p/=2;
    return sol;
}

int n,m, a[300000][50],q,p;
int main()
{
    f>>n>>m;
    int t=log2(n)+1;
    for (int i=1;i<=n;i++)
        f>>a[i][0];
    int o=1;
    for (int j=1;j<=t;j++)
    {
        o<<=1;
        for (int i=1;i<=n;i++)
        {
            a[i][j]= a[a[i][j-1]][j-1];
            if (a[i][j]==0&&r[i]==0) r[i]=o;
        }
    }
    for (int i=1;i<=m;i++)
    {
        f>>q>>p;
        if (p>r[q]||p>=n) g<<0<<'\n';
        else
        {
            while (p)
            {
                int x=0,y=p;
                while (y>1) x++,y>>=1;
                q=a[q][x];
                int s=1;
                while (s<=p) s<<=1;
                s>>=1;
                p-=s;
            }
            g<<q<<'\n';
        }
    }
    return 0;
}