Cod sursa(job #3239836)

Utilizator nicushor21Pirlog Marian Nicolae nicushor21 Data 7 august 2024 22:53:46
Problema Stramosi Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
int n,m,x,y,i,j,ans,v[250001];
vector<int> g[250001],st;
void dfs(int x){
    if(x != 0){
        v[x] = 0;
        dfs(g[x][0]);
        for(int i=st.size()-1;i>=0;i--)
            g[x].push_back(st[i]);
        st.push_back(g[x][0]);
    }
}
int main()
{
    st.clear();
    fin>>n>>m;
    for(i=1;i<=n;i++){
        fin>>v[i];
        g[i].push_back(v[i]);
    }
    for(i=n;i>0;i--){
        if(v[i] != 0){
            dfs(i);
            st.clear();
        }
    }
    while(m--){
        fin>>x>>y;
        j = g[x].size();
        y = min(j,y);
        fout<<g[x][y-1]<<'\n';
    }
    return 0;
}