Cod sursa(job #1061377)

Utilizator ard_procesoareLupicu ard_procesoare Data 19 decembrie 2013 17:43:35
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
#define NMAX 100005
#define pb push_back
vector <int> fiu[NMAX];
int t[NMAX],n,m,sol[NMAX],k[NMAX],str[NMAX],rad,niv[NMAX],nivel;
bool check[NMAX];
void read()
{
    int x,y;
    fin>>n;
    for(int i=1;i<=n;i++)
        fin>>k[i];
    for(int i=1;i<n;i++)
    {
        fin>>x>>y;
        t[y]=x;
        check[y]=true;
        fiu[x].pb(y);
    }
}
void dfs(int nod)
{
    //stramosul nodului:
    niv[++nivel] = nod;
    sol[nod] = 1 + sol[niv[nivel - k[nod]]];
    for(int i=0;i<fiu[nod].size();i++)
    {
        dfs(fiu[nod][i]);
    }
    nivel--;
}
int main()
{
    read();
    for(int i=1;i<=n;i++)
    {
        if(check[i] == false)
        {
            dfs(i);
            for(int i=1;i<=n;i++)
            fout<<sol[i]<<" ";
            return 0;
        }
    }
}