Cod sursa(job #3232536)

Utilizator StefanMilitaruMilitaru Stefan-Octavian StefanMilitaru Data 30 mai 2024 18:52:30
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <fstream>
#include <vector>
#include <iostream>

using namespace std;

int main()
{
    int n,m;
    ifstream fin("stramosi.in");
    ofstream fout("stramosi.out");
    fin >> n >> m;
    vector<vector<int>> stramosi;
    stramosi.resize(n + 1);
    stramosi[0].resize(18);
    for (int i = 1; i <= n; i++)
    {
        stramosi[i].resize(18);
        fin >> stramosi[i][0];
    }
    for(int i = 1; i <= 17; i++){
        for(int j = 1; j <= n; j++){
            stramosi[j][i] = stramosi[stramosi[j][i-1]][i - 1];
        }
    }
    for (int i = 1; i <= m; i++)
    {
        int x, y;
        fin >> x >> y;
        int putere = 0;
        while(y > 0){
            if(y %2 == 1){
                x = stramosi[x][putere];
            }
            putere += 1;
            y = y / 2;
        }
        fout << x << '\n';
    }
    return 0;
}