Pagini recente » Istoria paginii runda/urmasii_lui_moisil_2012_xi-xii/clasament | Cod sursa (job #1761838) | Cod sursa (job #1893417) | Cod sursa (job #2562056) | Cod sursa (job #971553)
Cod sursa(job #971553)
#include<fstream>
#include<vector>
#define pb push_back
#define maxn 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,m,root,u;
int v[maxn],t[maxn],sol[maxn],uz[maxn];
int stack[maxn];
vector <int> l[maxn];
void read()
{
int i;
int x,y;
fin>>n;
for(i=1;i<=n;i++) fin>>t[i];
for(i=1;i<n;i++)
{
fin>>x>>y;
uz[y]=1;
l[x].pb(y);
l[y].pb(x);
}
for(i=1;i<=n;i++) if(uz[i]==0) root=i;
}
void dfs(int k)
{
v[k]=1; stack[++u]=k;
if(t[k]) sol[k]=sol[stack[u-t[k]]]+1;
else sol[k]=0;
for(unsigned int i=0;i<l[k].size();i++)
if(v[l[k][i]]==0)
dfs(l[k][i]);
u--;
}
void print()
{
for(int i=1;i<=n;i++) fout<<sol[i]<<" ";
}
int main()
{
read();
dfs(root);
print();
fin.close();
fout.close();
return 0;
}