Cod sursa(job #2940302)

Utilizator AlexSerban21Serban Alexandru AlexSerban21 Data 15 noiembrie 2022 11:23:25
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin ("cerere.in");
ofstream fout ("cerere.out");
int n,i,x,y,t[100005],nn,sol[100005],st[100005],s[100005];
vector <int> v[100005];
void dfs (int nod,int niv)
{
    st[niv]=nod;
    if (s[nod]==0)
        sol[nod]=0;
    else
        sol[nod]=sol[st[niv-s[nod]]]+1;
    for (int i=0; i<v[nod].size (); i++)
        dfs (v[nod][i],niv+1);
}
int main()
{
    fin>>n;
    for (i=1; i<=n; i++)
        fin>>s[i];
    for (i=1; i<n; i++)
    {
        fin>>x>>y;
        v[x].push_back (y);
        t[y]=x;
    }
    for (i=1; i<=n; i++)
    {
        if (t[i]==0)
        {
            nn=i;
            break;
        }
    }
    dfs (nn,1);
    for (i=1; i<=n; i++)
        fout<<sol[i]<<" ";
    return 0;
}