Cod sursa(job #2139192)

Utilizator LauraNaduLaura Nadu LauraNadu Data 22 februarie 2018 10:59:19
Problema Cerere Scor 65
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.76 kb
#include<fstream>
#include<vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n, k[100002], d[100002], s[100002], h, rad;
bool viz[100002];
vector<int> l[100002];
void dfs(int nod)
{
    s[++h]=nod;
    viz[nod]=1;
    if(k[nod]!=0)
        d[nod]=d[s[h-k[nod]]]+1;
    for(int i=0;i<l[nod].size();i++)
        if(viz[l[nod][i]]==0)
            dfs(l[nod][i]);
    h--;
}
int main()
{
    f>>n;
    for(int i=1;i<=n;i++)
    {
        f>>k[i];
        if(rad==0 && k[i]==0)
            rad=i;
    }
    for(int i=1;i<n;i++)
    {
        int x, y;
        f>>x>>y;
        l[x].push_back(y);
        l[y].push_back(x);
    }
    dfs(rad);
    for(int i=1;i<=n;i++)
        g<<d[i]<<" ";
    return 0;
}