Cod sursa(job #2052424)

Utilizator AlexandruabcdeDobleaga Alexandru Alexandruabcde Data 30 octombrie 2017 16:33:41
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.66 kb
#include <fstream>

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