Cod sursa(job #1663359)

Utilizator andreimdvMoldovan Andrei andreimdv Data 25 martie 2016 21:22:19
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1 kb
#include <fstream>
#include<vector>
#define maxn 400001
using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");


int n,m,i,j,t,a,b;
int tata[maxn];
vector<int> v[maxn];

vector<int> querry[maxn];
vector<int> querrypos[maxn];

int sol[maxn],st[maxn];

void dfs(int nod,int level)
{
    int i;
    st[level]=nod;
    for(i=0;i<querry[nod].size();++i)
    {
        if(level>querry[nod][i]&&level-querry[nod][i]>0)
            sol[querrypos[nod][i]]=  st[level- querry[nod][i]];
    }
    for(i=0;i<v[nod].size();++i)
        dfs(v[nod][i],level+1);
}


int main()
{
    fin>>n>>m;
    for(i=1;i<=n;++i)
    {
        fin>>tata[i];
        v[tata[i]].push_back(i);
    }
    for(i=1;i<=m;++i)
    {
        fin>>a>>b;
        querry[a].push_back(b);
        querrypos[a].push_back(i);
    }
    for(i=1;i<=n;++i)
    {
        if(tata[i]==0)
            dfs(i,1);
    }

    for(i=1;i<=m;++i)
        fout<<sol[i]<<'\n';


    return 0;
}