Pagini recente » Cod sursa (job #971588) | Cod sursa (job #153222) | Cod sursa (job #509558) | Cod sursa (job #927657) | Cod sursa (job #3352248)
///Varianta stramba(de testare)
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int Nmax = 1e5 + 5;
vector <int> vecini[Nmax];
int k[Nmax], ans[Nmax], nivel[Nmax], hasParent[Nmax];
void dfs(int nod, int tata, int lvl )
{
nivel[lvl] = nod;
if ( k[nod] == 0 )
ans[nod] = 0;
else
ans[nod] = ans[ nivel[ lvl - k[nod] ] ] + 1;
for ( int vecin : vecini[nod] )
{
if ( vecin == tata )
continue;
dfs( vecin, nod, lvl + 1);
}
}
int main()
{
int n, i;
fin >> n;
for ( i = 1; i <= n; ++i )
fin >> k[i];
for ( i = 1; i <= n - 1; ++i )
{
int a, b;
fin >> a >> b;
vecini[a].push_back(b);
vecini[b].push_back(a);///aici
hasParent[b] = true;
}
int root;
for ( i = 1; i <= n; ++i )
if ( hasParent[i] == false )
root = i;
dfs(root, -1, 1);
for ( i = 1; i <= n; ++i )
fout << ans[i] << " ";
return 0;
}