Cod sursa(job #2499216)

Utilizator DariusDCDarius Capolna DariusDC Data 25 noiembrie 2019 17:56:41
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.96 kb
#include <bits/stdc++.h>
#define nmax 250005
#define mmax 300005

using namespace std;

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

int n, Q;
int tt[nmax];
int p, q;

int dp[19][nmax]; //dp k, i  ==> Stramosul 2^k al nodului i

inline void read()
{
    fin >> n >> Q;
    for (int i = 1; i <= n; i++)
        fin >> dp[0][i];
}

inline void build()
{
    int ok = 1;
    for (int k = 1; ok ; k++)
    {
        ok = 0;
        for (int i = 1; i <= n; i++)
        {
            dp[k][i] = dp[k-1][dp[k-1][i]];
            if (dp[k][i])
                ok = 1;
        }
    }
}

inline void solve()
{
    int k = 0;
    while (q != 0)
    {
       // cout << p << " ";
        if (q % 2 == 1)
            p = dp[k][p];
        k++;
        q /= 2;
    }
    fout << p << "\n";
}

int main()
{
    read();
    build();
    while (Q--)
    {
        fin >> p >> q;
        solve();
    }
    return 0;
}