Cod sursa(job #3338152)

Utilizator Alex_at_gameIustin-Alexandru Frateanu Alex_at_game Data 31 ianuarie 2026 22:53:17
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.3 kb
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define all(v) begin(v), end(v)
#define al(v, l, r) begin(v) + l, begin(v) + r + 1
#define pb push_back
#define pob pop_back
#define sz(v) (int)v.size()
#define fs first
#define sd second

constexpr int inf = 2e9;
constexpr ll infll = 4e18;
constexpr int N = 25e4 + 5;
constexpr int LOG = 19;

int n, m, p[N], rad, up[LOG][N];
vector<int> g[N];
signed main() {
    ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
    freopen("stramosi.in", "r", stdin);
    freopen("stramosi.out", "w", stdout);

    cin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        cin >> p[i];

        if (p[i] == 0) {
            rad = i;
        }

        g[p[i]].pb(i);
        g[i].pb(p[i]);
    }

    for (int i = 1; i <= n; ++i) {
        up[0][i] = p[i];
    }

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

    for (int i = 0, p, q; i < m; ++i) {
        cin >> q >> p;

        for (int k = LOG - 1; k >= 0; --k) {
            if ((p >> k) & 1) {
                q = up[k][q];
            }

            if (q == 0) {
                break;
            }
        }

        cout << q << "\n";
    }
}