Cod sursa(job #1789487)

Utilizator GeanaVladGeana Vlad GeanaVlad Data 27 octombrie 2016 01:19:20
Problema Stramosi Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.12 kb
#include <iostream>
#include<cstdio>
using namespace std;
int n,m,i,v[250001],p,k;
#define DIM 10000
char buff[DIM];
int poz=0;

void citeste(int &numar)
{
     numar = 0;
     //cat timp caracterul din buffer nu e cifra ignor
     while (buff[poz] < '0' || buff[poz] > '9')
          //daca am "golit" bufferul atunci il umplu
          if (++poz == DIM)
               fread(buff,1,DIM,stdin),poz=0;
     //cat timp dau de o cifra recalculez numarul
     while ('0'<=buff[poz] && buff[poz]<='9')
     {
          numar = numar*10 + buff[poz] - '0';
          if (++poz == DIM)
               fread(buff,1,DIM,stdin),poz=0;
     }
}
int stramos(int p,int k)
{
    bool ok=1;
    int st=v[p];
    for(int i=1;i<k && ok;i++)
    {
        st=v[st];
        if(!st)
            {ok=0;return ok;}
    }
    return st;
}
int main()
{
    freopen("stramosi.in","r",stdin);
    freopen("stramosi.out","w",stdout);
    citeste(n);
    citeste(m);
    for(i=1;i<=n;i++)
        citeste(v[i]);
    for(i=1;i<=m;i++)
    {
        citeste(p);
        citeste(k);
        cout<<stramos(p,k)<<'\n';
    }
}