Cod sursa(job #2474561)

Utilizator divianegoescuDivia Negoescu divianegoescu Data 15 octombrie 2019 15:32:47
Problema Cerere Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <fstream>
#include <vector>
#define DIM 100010
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,i,j,k[DIM],x,y,t[DIM],f[DIM],drum[DIM],sol[DIM],rad;
vector <int> v[DIM];
void dfs(int nod,int niv){
   f[nod]=1;
   drum[niv]=nod;
   sol[nod]=sol[drum[niv-k[nod]]]+1;
   for(int i=0;i<v[nod].size();i++){
       int vecin=v[nod][i];
        if(!f[vecin])
            dfs(vecin,niv+1);
    }
}

int main()
{
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>k[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]){
         rad=i;
         break;
       }
    dfs(rad,1);
    for(i=1;i<=n;i++)
        fout<<sol[i]-1<<" ";
    return 0;
}