Cod sursa(job #2474550)

Utilizator BogauuuBogdan Ivancu Bogauuu Data 15 octombrie 2019 15:22:50
Problema Cerere Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

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

vector <int> l[100005];

int d[100005],k[100005],af[100005];
bool v[100005];

void dfs(int nod, int niv)
{
    v[nod]=1;
    d[niv]=nod;
    int cm=d[niv];
    int cn=niv;
    int nr=0;
    while (k[cm]!=0)
    {
        cn-=k[cm];
        cm=d[cn];
        nr++;
    }
    af[nod]=nr;
    for (int i=0;i<l[nod].size();i++)
    {
        int vec=l[nod][i];
        if (v[vec]==0)
        {
            dfs(vec,niv+1);
        }
    }
}

int n,r,i,x,y;

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);
        l[y].push_back(x);
    }
    dfs(1,1);
    for (i=1;i<=n;i++) fout << af[i] << " ";

    return 0;
}