Cod sursa(job #2913773)

Utilizator biancalautaruBianca Lautaru biancalautaru Data 16 iulie 2022 21:21:04
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <fstream>
#include <vector>
#define DIM 100001
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,i,x,y,r,k[DIM],t[DIM],s[DIM],sol[DIM];
vector<int> l[DIM];

void dfs(int nod,int niv) {
    s[niv]=nod;
    if (k[nod]==0)
        sol[nod]=0;
    else
        sol[nod]=1+sol[s[niv-k[nod]]];
    for (int i=0;i<l[nod].size();i++)
        dfs(l[nod][i],niv+1);
}

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