Cod sursa(job #1303150)

Utilizator horiainfoTurcuman Horia horiainfo Data 27 decembrie 2014 17:45:46
Problema Stramosi Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <fstream>
#include <cstdio>
#include <vector>
using namespace std;
ofstream fout("stramosi.out");
vector<int> s[250001];
int p,q,m,n,a[250001];
void merge(int poz)
{
    if(a[poz]==0) return;
    else
    {
        s[poz].push_back(a[poz]);
        merge(a[poz]);
        for(int i=0;i<s[a[poz]].size();i++)
            s[poz].push_back(s[a[poz]][i]);
        a[poz]=0;
    }
}
int main()
{
    freopen("stramosi.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
        merge(i);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&q,&p);
        if(s[q].size()<p) fout<<0<<'\n';
        else fout<<s[q][p-1]<<'\n';
    }
    return 0;
}