Pagini recente » Cod sursa (job #1292059) | Cod sursa (job #2059484) | Cod sursa (job #504485) | Cod sursa (job #2281792) | Cod sursa (job #1042971)
#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];
/// dp = numarul de stramosi in sus al unui nod prin care trebuie sa treaca
/// deep = la al catelea nod de la radacina in jos ma aflu
/// 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;
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);
}
dfs(1,1);
for(int i=1;i<=n;++i)
g<<dp[i]<<" ";
return 0;
}