Cod sursa(job #1013162)

Utilizator cat_red20Vasile Ioana cat_red20 Data 20 octombrie 2013 14:39:32
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.91 kb
#include<fstream>
#include<vector>
#define MAXN 100001
using namespace std;
int n,back[MAXN],sol[MAXN],p[MAXN],st[MAXN],u;
vector<int>v[MAXN];
ifstream fin("cerere.in");
ofstream fout("cerere.out");

void read()
{
    int x,y;
    fin>>n;
    for(int i=1;i<=n;i++)
        fin>>back[i];
    for(int i=1;i<n;i++)
    {
        fin>>x>>y;
        v[x].push_back(y);
        p[y]=x;
    }
}

void dfs(int nod)
{
    if(back[nod]!=0)
        sol[nod]=sol[st[u-back[nod]]]+1;
    for(int i=0;i<v[nod].size();i++)
    {
        st[++u]=v[nod][i];
        dfs(v[nod][i]);
    }
    u--;
}

void print()
{
    for(int i=1;i<=n;i++)
    {
        fout<<sol[i]<<" ";
    }
}

int main()
{
    read();
    for(int i=1;i<=n;i++)
    {
        if(p[i]==0)
        {
            st[1]=i;
            u=1;
            dfs(i);
            break;
        }
    }
    print();
    return 0;
}