Cod sursa(job #2955211)

Utilizator alessiamtr12Mitrica Alessia alessiamtr12 Data 16 decembrie 2022 16:09:50
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <fstream>
#define NMAX 100001
#include<vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,x,y,m[NMAX],sol[NMAX],st[NMAX],rad;
bool viz[NMAX];
vector<int>v[NMAX];
void dfs(int nod,int niv)
{
    viz[nod]=1;
    st[niv]=nod;
    sol[nod]=sol[st[niv-m[nod]]]+1;
    for(int i=0;i<v[nod].size();i++)
    {
        int nod1=v[nod][i];
        if(viz[nod1]==0)
            dfs(nod1,niv+1);
    }
}
int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
        fin>>m[i];
    for(int i=1;i<n;i++)
    {
        fin>>x>>y;
        v[x].push_back(y);
        viz[y]=1;
    }
    for(int i=1;i<=n;i++)
    {
        if(viz[i]==0)
            rad=i;
        viz[i]=0;
    }
    dfs(rad,1);
    for(int i=1;i<=n;i++)
        fout<<sol[i]-1<<" ";
    return 0;
}