Cod sursa(job #2780773)

Utilizator DariusGhercaDarius Gherca DariusGherca Data 7 octombrie 2021 20:00:15
Problema Cerere Scor 85
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
const int N=1e5+10;
vector <int> a[N];
int tata[N],n,val[N],rez[N],stramos[N];
void calc(int x,int dist)
{
    stramos[dist]=x;
    rez[x]=rez[stramos[dist-val[x]]]+1;
    for(auto i:a[x])
    {
        calc(i,dist+1);
    }
}
int main()
{
    ios_base::sync_with_stdio(false);
    f>>n;
    for(int i=1;i<=n;i++)
    {
        f>>val[i];
        rez[i]=-1;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        f>>x>>y;
        tata[y]=x;
        a[x].push_back(y);
    }
    for(int i=1;i<=n;i++)
    {
        if(tata[i]==0)
        {
            calc(i,0);
        }
    }
    for(int i=1;i<=n;i++)
    {
        g<<rez[i]<<" ";
    }
    return 0;
}