Cod sursa(job #1735747)

Utilizator liviu23Liviu Andrei liviu23 Data 30 iulie 2016 20:21:10
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <fstream>
#include <vector>
#define DIM 250005
using namespace std;

int n,m,vf[DIM],st[DIM],ans[300005];
vector<int> adj[DIM];
vector<pair<int,int> > q[DIM];

void dfs(int p) {
    st[++st[0]]=p;
    for(int i=0;i<q[p].size();i++) {
        if(st[0]-q[p][i].first<=0)
            ans[q[p][i].second]=0;
        else
            ans[q[p][i].second]=st[st[0]-q[p][i].first];
    }
    for(int i=0;i<adj[p].size();i++)
        dfs(adj[p][i]);
    st[0]--;
}

int main()
{
    ifstream fin("stramosi.in");
    ofstream fout("stramosi.out");
    fin>>n>>m;
    int a,b;
    for(int i=1;i<=n;i++) {
        fin>>a;
        if(a==0)
            vf[++vf[0]]=i;
        else
            adj[a].push_back(i);
    }
    for(int i=1;i<=m;i++) {
        fin>>a>>b;
        q[a].push_back({b,i});
    }
    for(int i=1;i<=vf[0];i++)
        dfs(vf[i]);
    for(int i=1;i<=m;i++)
        fout<<ans[i]<<'\n';
    return 0;
}