Cod sursa(job #2212293)

Utilizator roberttrutaTruta Robert roberttruta Data 13 iunie 2018 19:01:53
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <fstream>
#include <vector>
using namespace std;
vector <int> v[100002];
int n,k,K,a,b,tati[100002],val[100002],cere[100002];
void dfs(int start)
{
    int i;
    for(i=0;i<v[start].size();i++)
    {
        k=v[start][i];
        if(cere[k])
        {
            K=k;
        while(cere[k])
        {
            K=tati[K];
            cere[k]--;
        }
        val[k]=val[K]+1;
        }
        dfs(k);
    }
}
int main()
{
    ifstream f("cerere.in");
    ofstream g("cerere.out");
int i;
    f>>n;
    for(i=1;i<=n;i++)
        f>>cere[i];
    for(i=2;i<=n;i++)
    {
        f>>a>>b;
        v[a].push_back(b);
        tati[b]=a;
    }
    dfs(1);
    for(i=1;i<=n;i++)
        g<<val[i]<<' ';
    return 0;
}