Cod sursa(job #2052428)

Utilizator AlexandruabcdeDobleaga Alexandru Alexandruabcde Data 30 octombrie 2017 16:35:05
Problema Cerere Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <fstream>

using namespace std;
ifstream f ("cerere.in");
ofstream g ("cerere.out");
int n,a[100002],lista[100002],x,y,afis,i,nr,j;
int main()
{
    f>>n;
    for (i=1;i<=n;i++) {
        f>>a[i];
    }
    for (i=1;i<=n-1;i++) {
        f>>x>>y;
        lista[y]=x;
    }
    for (i=1;i<=n;i++) {
        if (a[i]==0) g<<"0 ";
        else {
            nr=lista[i];
            for (j=1;j<=a[i]-1;j++) {
                nr=lista[nr];
            }
            afis=1;
            while (a[nr]!=0)
            {
                nr=lista[nr];
                afis++;
            }
            g<<afis<<" ";
        }
    }
    return 0;
}