Cod sursa(job #1744757)

Utilizator Bodo171Bogdan Pop Bodo171 Data 20 august 2016 13:25:02
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>
using namespace std;
const int nmax=100005;
int k[nmax],st[nmax],nr[nmax],a,b,top,n,i,node;
vector<int> v[nmax];
bitset<nmax> noroot;
void dfs(int x)
{
    top++;
    st[top]=x;
    if(k[x]==0)nr[x]=0;
    else nr[x]=nr[st[top-k[x]]]+1;
    for(int i=0;i<v[x].size();i++)
        dfs(v[x][i]);
    top--;
}
int main()
{
    ifstream f("cerere.in");
    ofstream g("cerere.out");
    f>>n;
    for(i=1;i<=n;i++)
        f>>k[i];
    for(i=1;i<=n-1;i++)
    {
        f>>a>>b;
        v[a].push_back(b);
        noroot[b]=1;
    }
    for(i=1;i<=n;i++)
        if(noroot[i]==0)
    {
        node=i;
        break;
    }
    dfs(node);
    for(i=1;i<=n;i++)
        g<<nr[i]<<' ';
    return 0;
}