Cod sursa(job #643027)

Utilizator okros_alexandruOkros Alexandru okros_alexandru Data 2 decembrie 2011 19:30:24
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.02 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() {
	unsigned int i,nod=stack[nr];
	while(nr) {
		if(!viz[nod])
			for(i=0;i<query[nod].size();i++) {
				stramos=query[nod][i].first;
				if(nr-stramos)
					sol[query[nod][i].second]=stack[nr-stramos];
				}
		if(viz[nod]<v[nod].size())
			stack[++nr]=v[nod][viz[nod]++];
		else
			--nr,viz[nod]++;
		nod=stack[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]) {
			stack[++nr]=i;
			dfs();
			}
	afis();
	return 0;
}