Mai intai trebuie sa te autentifici.
Cod sursa(job #1897929)
Utilizator | Data | 1 martie 2017 19:16:53 | |
---|---|---|---|
Problema | Cerere | Scor | 0 |
Compilator | cpp | Status | done |
Runda | Arhiva de probleme | Marime | 0.74 kb |
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> G[100005];
int N,dp[100005],st[100005],nst,K[100005];
void Dfs(int nod)
{
if(!K[nod])
dp[nod]=0;
else
dp[nod]=dp[st[nst-K[nod]+1]]+1;
st[++nst]=nod;
for(int i=0;i<G[nod].size();i++)
{
int son=G[nod][i];
Dfs(son);
}
nst--;
}
int main()
{
fin>>N;
for(int i=1;i<=N;i++)
{
fin>>K[i];
}
for(int i=1;i<N;i++)
{
int x,y;
fin>>x>>y;
G[x].push_back(y);
}
Dfs(1);
for(int i=1;i<=N;i++)
{
fout<<dp[i]<<" ";
}
return 0;
}