Cod sursa(job #1042934)

Utilizator ShaDoWsiD100Rzv Rzv ShaDoWsiD100 Data 27 noiembrie 2013 20:03:45
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <fstream>
#include<vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n,i,nivel[100001],sol[100001],v[100001],x,y,t[100001];
vector <int> a[100001];
void df(int nod,int nv){
    nivel[nv]=nod;
    if(v[nod]!=0)
        sol[nod]=sol[nivel[nv-v[nod]]]+1;
    for(int i=0;i<a[nod].size();i++)
        df(a[nod][i],nv+1);

}
int main()
{
    f>>n;
    for(i=1;i<=n;i++)
        f>>v[i];
    for(i=1;i<n;i++)
    {
        f>>x>>y;
        a[x].push_back(y);
        t[y]=x;
    }
    for(i=1;i<=n;i++)
        if(t[i]==0){
            df(i,1);
            break;}
    for(i=1;i<=n;i++)
        g<<sol[i]<<" ";

}