Cod sursa(job #2890138)

Utilizator matthriscuMatt . matthriscu Data 14 aprilie 2022 17:56:30
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.69 kb
#include <bits/stdc++.h>
using namespace std;

#define NMAX 250005
#define LOGMAX 20

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

	int n, m;
	fin >> n >> m;

	vector<int> log(n + 1);
	log[1] = 0;
	for (int i = 2; i <= n; ++i)
		log[i] = log[i / 2] + 1;

	vector<vector<int>> dp(log[n] + 1, vector<int>(n + 1, 0));
	for (int i = 1; i <= n; ++i)
		fin >> dp[0][i];

	for (int i = 1; i <= log[n]; ++i)
		for (int j = 1; j <= n; ++j)
			dp[i][j] = dp[i - 1][dp[i - 1][j]];

	while (m--) {
		int p, q;
		fin >> q >> p;

		int ans = 0, pow;
		for (pow = 1; pow <= p; pow <<= 1);
		for (; pow; pow >>= 1)
			if (p & pow)
				ans = q = dp[log[pow]][q];

		fout << ans << '\n';
	}
}