Pagini recente » Cod sursa (job #1958020) | Cod sursa (job #2138972)
#include <fstream>
#include <vector>
using namespace std;
ifstream fin ("cerere.in");
ofstream fout ("cerere.out");
vector <int> L[100010];
int n,i,x,y,f[100010],v[100010],k,K[100010],sol[100010],rad;
//simulezi intr-un vector dfs
// 1
// 2 5
// 3 4 6
//dfs de duce i 2 3 apoi scoate pe 3 si merge in 4 (strmosii tot 1 si 2 raman) etc
//deci sol i este sol elementului al Ki-lea de la coada
//sol se calculeaza de la rad spre frunze deci sol de stramos de i e deja calculat!
void dfs(int nod){
if(K[nod]>0)
sol[nod]=1+sol[v[k-K[nod]+1]];
v[++k]=nod;
for(int i=0;i<L[nod].size();i++)
dfs(L[nod][i]);
--k;
}
int main(){
fin>>n;
for(i=1;i<=n;i++)
fin>>K[i];
for(i=1;i<n;i++){
fin>>x>>y;
L[x].push_back(y);
f[y]=1;
}
for(i=1;i<=n;i++)
if(f[i]==0){
rad=i;
break;
}
dfs(rad);
for(i=1;i<=n;i++)
fout<<sol[i]<<" ";
}