Cod sursa(job #3237268)

Utilizator Sorin_GabrielGabara Sorin Gabriel Sorin_Gabriel Data 7 iulie 2024 21:23:50
Problema Stramosi Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <fstream>
#include <vector>
#define VMAX 250004
using namespace std;
ifstream fin ("stramosi.in");
ofstream fout ("stramosi.out");

vector<int> stramosi[VMAX];


int main()
{
    fin.tie(0);
    int n,m,i,j,k,q,nr;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>j;
        stramosi[i].push_back(j);
        for(j=1,k=0;i>j;j*=2,k++)
        {
            stramosi[i].push_back(stramosi[i-j][k]);
        }
    }
    while(m--)
    {
        fin>>q>>nr;
        k=0;

        while(nr)
        {
            while(nr%2==0)
                    {
                        k++;
                        nr/=2;
                    }

            if(k<stramosi[q].size())
            q=stramosi[q][k];
            else
            {
                fout<<"0\n";
                break;
            }
            nr--;
        }
        if(nr==0)
        fout<<q<<'\n';
    }

    return 0;
}