Cod sursa(job #643019)

Utilizator okros_alexandruOkros Alexandru okros_alexandru Data 2 decembrie 2011 19:21:29
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.92 kb
#include<fstream>
#include<vector>
#define NMAX 250500
#define NMAX2 300300
using namespace std;
vector <int> v[NMAX];
vector <pair<int,int> > query[NMAX];
int n,sol[NMAX2],stack[NMAX],nr,q,viz[NMAX],stramos;
void afis() {
	int i;
	ofstream out("stramosi.out");
	for(i=0;i<q;i++)
		out<<sol[i]<<'\n';
	out.close();
}
void dfs(int nod) {
	unsigned int i;
	stack[++nr]=nod;
	viz[nod]=1;
	for(i=0;i<query[nod].size();i++) {
		stramos=query[nod][i].first;
		if(nr-stramos)
			sol[query[nod][i].second]=stack[nr-stramos];
		}
	for(i=0;i<v[nod].size();i++)
		dfs(v[nod][i]);
	--nr;
}
void citire() {
	int i,x,y;
	ifstream in("stramosi.in");
	in>>n>>q;
	for(i=0;i<n;i++) {
		in>>x;
		v[x].push_back(i+1);
		}
	for(i=0;i<q;i++) {
		in>>x>>y;
		query[x].push_back(pair<int,int>(y,i));
		}
	in.close();
}
int main() {
	citire();
	for(int i=1;i<=n;i++)
		if(!viz[i])
			dfs(i);
	afis();
	return 0;
}