Pagini recente » Cod sursa (job #497537) | Cod sursa (job #26571) | Cod sursa (job #2466707) | Cod sursa (job #1863030) | Cod sursa (job #1069600)
#include<fstream>
#include<vector>
#define Nmax 100001
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> V[Nmax];
int sol[Nmax], K[Nmax], root;
int st[Nmax];
void dfs(int nod, int k)
{
st[k] = nod;
if (K[nod])
sol[nod] = sol[st[k - K[nod]]] + 1;
for (int i = 0; i < V[nod].size(); ++i)
dfs(V[nod][i], k + 1);
}
int main()
{
int x, y, N, i;
fin >> N;
for (i = 1; i <= N; ++i)
{
fin >> K[i];
if (K[i])
sol[i] = 1;
}
root = N;
for (i = 1; i < N; ++i)
{
fin >> x >> y;
V[x].push_back(y);
root ^= y ^ i;
}
dfs(root, 0);
for (i = 1; i <= N; ++i)
fout << sol[i] << ' ';
fout << '\n';
fin.close();
fout.close();
return 0;
}