Cod sursa(job #3239848)

Utilizator nicushor21Pirlog Marian Nicolae nicushor21 Data 8 august 2024 00:02:11
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.93 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],cnt;
pair<int,int> sol[250001];
vector<int> g[250001];
void dfs(int x){
    g[cnt].push_back(x);
    if(!sol[x].first) sol[x].first = cnt,sol[x].second = g[cnt].size()-1;
    if(x != 0){
        dfs(v[x]);
    }
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=n;i++){
        fin>>v[i];
    }
    for(i=n;i>0;i--)
        if(!sol[i].first){
            cnt++;
            dfs(i);
        }
    while(m--){
        fin>>x>>y;
        y = min(sol[x].second+y,(int)g[sol[x].first].size()-1);
        fout<<g[sol[x].first][y]<<'\n';
    }
    /*
    for(i=1;i<=cnt;i++){
        cout<<i<<": ";
        for(j=0;j<g[i].size();j++)
            cout<<g[i][j]<<' ';
        cout<<'\n';
    }
    for(i=1;i<=n;i++)
        cout<<sol[i].first<<','<<sol[i].second<<' ';
    */
    return 0;
}