Cod sursa(job #1867869)

Utilizator Mircea_DonciuDonciu Mircea Mircea_Donciu Data 4 februarie 2017 13:24:10
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.06 kb
#include<fstream>

using namespace std;
int n,nr,pp,i,j,x,y,k,vf[200005],lst[100005],pred[100005],urm[100005],v[100005],sol[100005];
bool viz[100005],c[100005];
void adauga(int x, int y)
{
    nr++;
    vf[nr]=y;
    urm[nr]=lst[x];
    lst[x]=nr;
}
void dfs(int x)
{
    viz[x]=true;
    pred[pp]=x;
    int poz=lst[x],y,i,k;
    if(v[x]==0)
        sol[x]=0;
    else
    {
        k=x;
        k=pred[pp-v[x]];
        sol[x]=sol[k]+1;
    }
    while(poz!=0)
    {
        y=vf[poz];
        if(!viz[y])
        {
            pp++;
            dfs(y);
            pp--;
        }
        poz=urm[poz];
    }
}
int main()
{
    ifstream f("cerere.in");
    ofstream g("cerere.out");
    f>>n;
    for(i=1; i<=n; i++)
        f>>v[i];
    for(i=1; i<=n-1; i++)
    {
        f>>x>>y;
        adauga(x,y);
        c[y]=true;
    }
    for(i=1; i<=n; i++)
        if(c[i]==false)
        {
            pp=1;
            dfs(i);
            break;
        }
    for(i=1; i<=n; i++)
        g<<sol[i]<<" ";
    return 0;
}