Cod sursa(job #3225590)

Utilizator Zed1YasuoAlex Birsan Zed1Yasuo Data 18 aprilie 2024 01:25:29
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.14 kb
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
ifstream f("stramosi.in");
ofstream g("stramosi.out");
int n,m,k=1,viz[250003],st[250003];
int detect[300003];
struct punct
{
    int val,cur;
};
vector<punct>quiz[250003];
vector<int>a[250003];
void dfs(int nod,int nivel)
{
    st[nivel]=nod;
    viz[nod]=1;
    if(quiz[nod].size())
        for(int i=0;i<quiz[nod].size();i++)
            if(nivel<=quiz[nod][i].val)
                detect[quiz[nod][i].cur]=0;
            else
                detect[quiz[nod][i].cur]=st[nivel-quiz[nod][i].val];
    for(int i=0;i<a[nod].size();i++)
    {
        int vecin=a[nod][i];
        if(!viz[vecin])
        {
            dfs(vecin,nivel+1);
        }
    }
}
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[x].push_back({y,i});
    }
    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;
}