Cod sursa(job #3133768)

Utilizator johnutddDobrin Ionut johnutdd Data 26 mai 2023 20:29:12
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>
#include <vector>

using namespace std;

ifstream f("stramosi.in");
ofstream g("stramosi.out");

void construireMatrice(int n, vector<vector<int>>& a)
{
    for (int j = 1; j <= 15; j++)
        for (int i = 1; i <= n; i++)
            a[i][j] = a[a[i][j - 1]][j - 1];
}

void intrebare(int z, int &k, vector<vector<int>>& a)
{
    for (int j = 15; j >= 0; j--)
        if ((z >> j) & 1)
            k = a[k][j];
}

int main()
{
    int n, m, z, k, i;
    f >> n >> m;
    vector<vector<int>> a(n + 1, vector<int>(16)); 

    for (i = 1; i <= n; i++)
        f >> a[i][0];

    construireMatrice(n, a);

    for (i = 1; i <= m; i++)
    {
        f >> k >> z;
        intrebare(z, k, a);
        g << k << '\n';
    }

    return 0;
}