Cod sursa(job #2801039)

Utilizator CelestinNegraru Celestin Celestin Data 14 noiembrie 2021 18:48:11
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.97 kb
#include <fstream>
#include <vector>
#include <unordered_set>
#define maxi 350000
using namespace std;
vector<int> V[maxi+10];
vector<pair<int,int>> G[maxi+10];
unordered_set<int> radacini;
bool viz[maxi+10];
int nod[maxi+10],ans[maxi+10],n,m,x,y;
void DFS(int x,int lvl)
{
    viz[x]=true;
    nod[lvl]=x;
    for(auto a:G[x])
    {
        if(lvl-a.first>=0)
        {
            ans[a.second]=nod[lvl-a.first];
        }
    }
    for(auto a:V[x])
    {
        if(!viz[a])
            DFS(a,lvl+1);
    }
}
int main()
{
    ifstream f("stramosi.in");
    ofstream g("stramosi.out");
    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        f>>x;
        if(x)
        {
            V[x].push_back(i);
        }
        else radacini.insert(i);
    }
    for(int i=1;i<=m;i++)
    {
        f>>x>>y;
        G[x].push_back(make_pair(y,i));
    }
   for(auto a:radacini)
    DFS(a,0);
   for(int i=1;i<=m;i++)
    g<<ans[i]<<'\n';
    return 0;
}