Cod sursa(job #2076987)

Utilizator dragos231456Neghina Dragos dragos231456 Data 27 noiembrie 2017 16:09:01
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int K[100005],n,cd[100005],k,v[100005],x,y,root;
vector<int> vecini[100005];

void dfs(int x)
{
    cd[++k]=x;
    v[x]=v[cd[k-K[x]]]+1;
    for(int i=0;i<vecini[x].size();++i)
    {
        dfs(vecini[x][i]);
    }
    --k; return;
}

int main()
{
    f>>n;
    for(int i=1;i<=n;++i)
    {
        f>>K[i];
        v[i]=-1;
    }
    for(int i=1;i<n;++i)
    {
        f>>x>>y;
        vecini[x].push_back(y);
        cd[y]=1;
    }
    for(int i=1;i<=n;++i)
    {
        if(cd[i]==0) root=i;
        cd[i]=0;
    }
    dfs(root);
    for(int i=1;i<=n;++i) g<<v[i]<<' ';
    return 0;
}