Cod sursa(job #2212335)

Utilizator roberttrutaTruta Robert roberttruta Data 13 iunie 2018 20:31:10
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>
#include <vector>
using namespace std;
vector <int> v[100002];
int n,k,a,b,I,tati[100002],val[100002],cere[100002],nivel[100002];
void dfs(int start, int loc)
{
    int i;
    nivel[loc]=start;
    for(i=0;i<v[start].size();i++)
    {
        k=v[start][i];
        if(cere[k])
        val[k]=val[nivel[loc-cere[k]+1]]+1;
        dfs(k,loc+1);
    }
}
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;
    }
    for(i=1;i<=n;i++)
        if(tati[i]==0)
    {
        I=i;
        break;
    }
    dfs(I,1);
    for(i=1;i<=n;i++)
        g<<val[i]<<' ';
    return 0;
}