Pagini recente » Cod sursa (job #267407) | Cod sursa (job #2411923) | Cod sursa (job #2369774) | Cod sursa (job #3252636) | Cod sursa (job #2798441)
#include <vector>
#include <fstream>
#include <bitset>
using namespace std;
const int N=100001;
vector<int> a[N];
int stram[N];
int stramos[N];
int cnt=0;
bitset<N>dif_radacina;
void parcurgere(int x){
stram[cnt]=x;
if(stramos[x]!=0){
int str=stram[cnt-stramos[x]];
stramos[x]=stramos[str]+1;
}
cnt++;
for(auto y:a[x]){
parcurgere(y);
}
cnt--;
}
int main() {
ios_base::sync_with_stdio(false);
ifstream in("cerere.in");
ofstream out("cerere.out");
int n;
in>>n;
for(int i=1;i<=n;i++){
in>>stramos[i];
}
int x,y;
for(int i=1;i<n;i++){
in>>x>>y;
a[x].push_back(y);
dif_radacina[y]=true;
}
int radacina;
for(int i=1;i<=n;i++){
if(!dif_radacina[i]){
radacina=i;
i=n+1;
}
}
parcurgere(radacina);
for(int i=1;i<=n;i++){
out<<stramos[i]<<" ";
}
}