Cod sursa(job #2955508)

Utilizator andrei_C1Andrei Chertes andrei_C1 Data 17 decembrie 2022 11:41:49
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.72 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");

const int NMAX = 25e4;
const int LGMAX = 18;

int n, m;
int par[LGMAX][NMAX + 1]; // par[i][j] =def= al (2^i)-lea stramos al lui j

int main() {
	fin >> n >> m;

	for(int i = 1; i <= n; i++) {
		fin >> par[0][i];
	}

	for(int i = 1; (1 << i) < n; i++) {
		for(int j = 1; j <= n; j++) {
			int k = par[i - 1][j];
			par[i][j] = par[i - 1][k];
		}
	}

	for(int tc = 1; tc <= m; tc++) {
		int q, p;
		fin >> q >> p;

		int stramos = q;
		for(int i = 0; (1 << i) <= p; i++) {
			if(p & (1 << i)) { // daca bitul 1 este setat (este egal cu 1) in p
				stramos = par[i][stramos];
			}
		}

		fout << stramos << '\n';
	}
	return 0;
}