Cod sursa(job #2955813)

Utilizator gabriel.9619Gabriel Stefan Tita gabriel.9619 Data 17 decembrie 2022 20:54:35
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int k[100001], t[100001], s[100001], sol[100001];
vector <int> l[100001];

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

int main()
{
    int n, i, x, y, xs;
    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)
        {
            xs=i;
            break;
        }
    }
    dfs(xs, 1);
    for(i=1;i<=n;i++)
    {
        fout<<sol[i]<<" ";
    }
}