Cod sursa(job #2421348)

Utilizator toadehuPuscasu Razvan Stefan toadehu Data 14 mai 2019 20:02:59
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <bits/stdc++.h>
using namespace std;

vector <int> graf[100005];

int n,v[100005],grad[100005],viz[100005],stivgraf[100005],ans[100005];

void dfs(int x,int nr)
{
    int i;
    stivgraf[nr]=x;
    if(v[x]!=0)
    {
        ans[x]=1+ans[stivgraf[nr-v[x]]];
    }
    viz[x]=1;
    for(i=0; i<graf[x].size(); i++)
    {
        if(viz[graf[x].at(i)]==0)
        {
            dfs(graf[x].at(i),nr+1);

        }
    }
}

int main()
{
    ifstream fin("cerere.in");
    ofstream fout("cerere.out");
    int i,x,y;
    fin>>n;
    for(i=1; i<=n; i++)
    {
        fin>>v[i];
    }
    for(i=1; i<=n; i++)
    {
        fin>>x>>y;
        grad[y]++;
        graf[x].push_back(y);
        graf[y].push_back(x);
    }
    for(i=1; i<=n; i++)
    {
        if(grad[i]==0)
        {
            dfs(i,1);
        }
    }
    for(i=1; i<=n; i++)
    {
        fout<<ans[i]<<" ";
    }
    return 0;
}