Cod sursa(job #3237265)

Utilizator Sorin_GabrielGabara Sorin Gabriel Sorin_Gabriel Data 7 iulie 2024 21:15:03
Problema Stramosi Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.92 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()
{
    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;
        for(j=1,k=0;j<=nr;j*=2,k++);

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

    return 0;
}