Cod sursa(job #391306)

Utilizator mathboyDragos-Alin Rotaru mathboy Data 5 februarie 2010 14:04:29
Problema Stramosi Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.05 kb
#include <fstream>
#include <vector>
#define DIM 300005
#define pb push_back
#define sz size

using namespace std;

ifstream fin ("stramosi.in"); // citire + scriere cu streamuri pt eficienta (cica)
ofstream fout ("stramosi.out");

int N, M, i, p, q , a;
int st [DIM];
vector <int> listv [DIM];
vector <int> listp [DIM], G [DIM];
int sol [350005], fat [DIM];
bool viz [DIM];

void dfs (int node, int lev) {
    
	int j;
	viz [node] = true;
	st [lev] = node;
	    for (j = 0; j < listv [node].sz (); j++) 
		    if (lev >= listv [node][j]) sol [listp [node][j]] = st [lev - listv [node][j]];
	for (j = 0; j < G [node].sz (); j++) 
		if (!viz [G [node][j]]) 
		dfs (G [node][j], lev + 1);
}

int main () {
	
	int a,t, j;
	fin >> N >> M;
	for (i = 1; i <= N; i++) {
		fin >> a;
		fat [i] = a;
		G [a].pb (i);
	}
    for (i = 1; i <= M; i++) 
	{
		fin >> q >> p;
		listv [q].pb (p);
		listp [q].pb (i);
	}
	for (i = 1; i <= N; i++)
		if (!fat [i])
			dfs (i, 0);
	for (i = 1; i <= M; i++) fout << sol [i] << "\n";
	return 0;
}