Cod sursa(job #1068858)

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

using namespace std;

vector<int> l[100005];

int n, i, a, b;
long long 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;
    }
    x=1LL*n*(n+1)/2;
    for(i=1;i<n;i++)
    {
        f>>a;f>>b;
        l[a].push_back(b);
        x-=b;
    }
    dfs((int)x,1);
    for(i=1;i<=n;i++)
        g<<v[i]<<" ";
    return 0;
}