Cod sursa(job #2139182)

Utilizator LauraNaduLaura Nadu LauraNadu Data 22 februarie 2018 10:55:20
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.69 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;
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];
    for(int i=1;i<n;i++)
    {
        int x, y;
        f>>x>>y;
        l[x].push_back(y);
        l[y].push_back(x);
    }
    dfs(1);
    for(int i=1;i<=n;i++)
        g<<d[i]<<" ";
    return 0;
}