Cod sursa(job #1856769)

Utilizator bt.panteaPantea Beniamin bt.pantea Data 25 ianuarie 2017 13:47:12
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <iostream>
#include <fstream>
#include <cmath>
using namespace std;
ifstream f ("stramosi.in");
ofstream g ("stramosi.out");
int r[300000];
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
        {
            for (int i = log2(p); i >= 0; i--)
                if ((1<<i) & p)
                    q = a[q][i];
            g<< q<<'\n';
        }
    }
    return 0;
}