Cod sursa(job #2953974)

Utilizator Zed1YasuoAlex Birsan Zed1Yasuo Data 12 decembrie 2022 20:08:25
Problema Stramosi Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.15 kb
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
ifstream f("stramosi.in");
ofstream g("stramosi.out");
int n,m,k=1,viz[250001],st[250001];
int detect[250001];
struct pct
{
    int val,tata,cur;
}quiz[300003];
vector<int>a[250001];
void dfs(int nod,int nivel)
{
    st[nivel]=nod;
    viz[nod]=1;
    if(nod==quiz[k].tata)
        detect[quiz[k].cur]=st[nivel-quiz[k].val],k++;
        //g<<nod<<" "<<st[nivel-quiz[k].val]<<'\n',k++;
    for(int i=0;i<a[nod].size();i++)
    {
        int vecin=a[nod][i];
        if(!viz[vecin])
        {
            dfs(vecin,nivel+1);
        }
    }
}
bool comp(pct X,pct Y)
{
    return X.tata<Y.tata;
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int x;
        f>>x;
        if(x)
            a[x].push_back(i);
    }
    for(int i=1;i<=m;i++)
    {
        int x,y;
        f>>x>>y;
        quiz[i].tata=x;
        quiz[i].val=y;
        quiz[i].cur=i;
    }
    sort(quiz+1,quiz+m+1,comp);
    for(int i=1;i<=n;i++)
        if(!viz[i])
            dfs(i,1);
    for(int i=1;i<=m;i++)
        g<<detect[i]<<'\n';
    return 0;
}