Cod sursa(job #1068855)

Utilizator iordache.bogdanIordache Ioan-Bogdan iordache.bogdan Data 28 decembrie 2013 21:01:27
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <fstream>
#include <vector>

using namespace std;

vector<int> l[100005];

int n, i, a, b, x;
int k[100005], v[100005], P[100005];
bool ok[100005], o[100005];

void dfs(int x, int niv)
{
    ok[x]=1;
    P[niv]=x;
    v[x]=v[P[niv-k[x]]]+1;
    for(int i=0;i<l[x].size();i++)
    {
        int y=l[x][i];
        if(!ok[y])
            dfs(y, niv+1);
    }

}

int main()
{
    ifstream f("cerere.in");
    ofstream g("cerere.out");
    f>>n;
    for(i=1;i<=n;i++)
    {
        f>>k[i];
        v[i]=-1;
    }
    for(i=1;i<n;i++)
    {
        f>>a;f>>b;
        l[a].push_back(b);
        l[b].push_back(a);
        o[b]=1;
    }
    for(i=1;i<=n;i++)
        if(!o[i])
        {
            x=i;
            break;
        }
    dfs(x,1);
    for(i=1;i<=n;i++)
        g<<v[i]<<" ";
    return 0;
}