Pagini recente » Cod sursa (job #708064) | Cod sursa (job #1381452) | Cod sursa (job #1983513) | Cod sursa (job #2072098) | Cod sursa (job #1042997)
#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>
#define DN 100005
#define next_nod list[nod][i]
using namespace std;
vector<int> list[DN];
vector<int>::iterator it;
bitset<DN> viz;
int deep[DN],dp[DN],s[DN],t[DN];
/// dp = numarul de stramosi in sus al unui nod prin care trebuie sa treaca
/// deep = nodul corespunzator unui nivel
/// s = la al catelea nod in sus trebuie sa merg pentru a rezovla
void dfs(int nod,int niv)
{
viz[nod]=true;
deep[niv]=nod;
if(s[nod])
dp[nod]=1 + dp [ deep[ niv - s[nod] ] ];
for(int i=0;i<list[nod].size();++i)
if(!viz[ next_nod ])
dfs( next_nod , niv+1);
}
int main()
{
int n,root;
ifstream f("cerere.in");
ofstream g("cerere.out");
f>>n;
for(int i=1;i<=n;++i)
f>>s[i];
for(int i=1;i<n;++i)
{
int a,b;
f>>a>>b;
list[a].push_back(b);
++t[b];
}
for(int i=1;i<=n;++i)
if(!t[i])
root=i;
dfs(root,1);
for(int i=1;i<=n;++i)
g<<dp[i]<<" ";
return 0;
}