Cod sursa(job #1695303)

Utilizator al.mocanuAlexandru Mocanu al.mocanu Data 26 aprilie 2016 21:27:15
Problema Stramosi Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.53 kb
#include <stdio.h>
#include <vector>
using namespace std;

const int MAX = 250005;
int n, m, x, y, p[MAX];
vector<int> l[MAX];

int main(){
	freopen("stramosi.in", "r", stdin);
	freopen("stramosi.out", "w", stdout);
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i++)
		scanf("%d", &p[i]);

	for(int i = 1; i <= n; i++){
		int x = i;
		while(x){
			x = p[x];
			l[i].push_back(x);
		}
	}

	for(int i = 0; i < m; i++){
		scanf("%d%d", &x, &y);
		if(y > l[x].size() + 1)
			printf("0\n");
		else
			printf("%d\n", l[x][y - 1]);
	}
	return 0;
}