Cod sursa(job #2890122)

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

int main()
{
	freopen("stramosi.in", "r", stdin);
	freopen("stramosi.out", "w", stdout);
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	int n, m;
	cin >> 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(n + 1, vector<int>(log[n] + 1, 0));

	for (int i = 1; i <= n; ++i)
		cin >> dp[i][0];
	
	for (int i = 1; i <= n; ++i)
		for (int j = 1; j <= log[n]; ++j)
			dp[i][j] = dp[dp[i][j - 1]][j - 1];
	
	while (m--) {
		int p, q;
		cin >> q >> p;

		int ans = 0;
		for (int i = 0; (1 << i) <= p; ++i)
			if (p & (1 << i))
				ans = q = dp[q][i];

		cout << ans << '\n';
	}
}