Cod sursa(job #390913)

Utilizator mathboyDragos-Alin Rotaru mathboy Data 4 februarie 2010 19:39:39
Problema Stramosi Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1 kb
#include <fstream>
#include <vector>
#define DIM 200000
#define pb push_back

using namespace std;

ifstream fin ("stramosi.in"); // citire + scriere cu streamuri pt eficienta (cica)
ofstream fout ("stramosi.out");
vector <int> G [DIM], sol [DIM];
int N, M, i, p, q, viz [DIM], a, st [DIM];
void dfs (int node, int lev) {
    
	int j;
	st [++lev] = node;
	for (j = 0; j < G [node].size (); j++) 
		if (!viz [G [node][j]]) {
			viz [G [node][j]] = 1;
			dfs (G [node][j], lev);
		}
	-- lev;
	for (int k = lev; k >= 1; k--)
		sol [node].pb (st [k]);
}

int main () {
	
	int j, t;
	fin >> N >> M;
	for (i = 1; i <= N; i++) {
		fin >> a;
		G [a].pb (i);
	}
	
	for (i = 0; i <= N; i++)
		if (!viz [i])
			dfs (i, 0);
	
	for (i = 1; i <= M; i++) {
		fin >> q >> p;
		for (t = 0, j = 0; j < sol [q].size (); j++) 
			if (j == p - 1) {
				fout << sol [q][j] << "\n";
				t = 1;
			}
			if (t == 0) fout << 0 <<"\n";
	}
	return 0;
}
//trimisesem o sursa gresita. Greseala mea.