Cod sursa(job #2357493)

Utilizator alexkosaAlex Kosa alexkosa Data 27 februarie 2019 14:36:26
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.76 kb
#include <fstream>
#include <vector>

using namespace std;

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

int n,viz[100003],st[100003],k[100003],sol[100003],dim;

vector<int>g[100003];

void dfs(int nod)
{
    st[++dim]=nod;
    int i=nod;
    if(k[i])
    {
        sol[i]=sol[st[dim-k[i]]]+1;
    }
    
  
    for(vector<int> :: iterator it=g[nod].begin();it!=g[nod].end();it++)
        dfs(*it);
    dim--;
}

int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
        fin>>k[i];
    for(int i=1;i<n;i++)
    {
        int x,y;
        fin>>x>>y;
        g[x].push_back(y);
        viz[y]++;
    }
    for(int i=1;i<=n;i++)
        if(viz[i]==0)
        {
            dfs(i);
            break;
        }
    for(int i=1;i<=n;i++)
        fout<<sol[i]<<" ";
}