Cod sursa(job #3253573)

Utilizator Bogdan345Marius Mihalache Bogdan345 Data 3 noiembrie 2024 14:39:35
Problema Stramosi Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("stramosi.in");
ofstream cout("stramosi.cout");
vector<vector<int>>dp;
vector<int>log_2;
const int val=17;
int main(){
    int n,m,nod,nr_stramos;
    cin>>n>>m;
    dp.resize(val+1,vector<int>(n+1));
    log_2.resize(n+1,0);

    for(int i=1;i<=n;i++){
        cin>>dp[0][i];
    }
    for(int i=2;i<=n;i++){
        log_2[i]=log_2[i/2]+1;
    }
  
    for(int j=1;j<=log_2[n];j++){
        for(int i=1;i<=n;i++){
            dp[j][i]=dp[j-1][dp[j-1][i]];
        }
    }

    while(m--){
        cin>>nod>>nr_stramos;
        while(nr_stramos){
            nod=dp[log_2[nr_stramos]][nod];
            nr_stramos-=1<<log_2[nr_stramos];
        }
        cout<<nod<<'\n';
    }

}