Pagini recente » Cod sursa (job #2945302) | Cod sursa (job #43840) | Cod sursa (job #787656) | Cod sursa (job #2022299) | Cod sursa (job #1398545)
#include <fstream>
#include <vector>
using namespace std;
#define NMax 100005
ifstream f("cerere.in");
ofstream g("cerere.out");
int N,K[NMax];
vector<int> v[NMax];
int st[NMax];
int sol[NMax];
void DFS(int x,int poz)
{
st[poz] = x;
for(int i=0;i<v[x].size();++i)
{
if(K[v[x][i]]) sol[v[x][i]] = sol[st[poz+1-K[v[x][i]]]]+1;
else sol[v[x][i]] = 0;
DFS(v[x][i],poz+1);
}
}
int tata[NMax];
int main()
{
int i,a,b;
f>>N;
for(i=1;i<=N;++i) f>>K[i];
int root = N*(N+1)/2;
for(i=1;i<N;++i)
{
f>>a>>b;
v[a].push_back(b);
root -= b;
}
DFS(root,1);
for(i=1;i<=N;++i) g<<sol[i]<<" "; g<<"\n";
f.close();
g.close();
return 0;
}