Pagini recente » Cod sursa (job #656582) | Cod sursa (job #849603) | Cod sursa (job #2032538) | Cod sursa (job #2536843) | Cod sursa (job #2916444)
#include <fstream>
#include <vector>
#include <algorithm>
#define NMAX 100010
using namespace std;
ifstream fin ("cerere.in");
ofstream fout ("cerere.out");
int n;
int nrstramos[NMAX];
int sol[NMAX];///sol[nod]=nr de cereri pt nod
vector <int> L[NMAX];///necesare pentru dfs L[nod]=fii lui nod(salvand in liste doar fii nu mai este necesar vectorul de vizitare)
int tata[NMAX];///necesar pentru a determina radacina, va fi nodul care nu are tata, adica tata[nod]=0;
int drum[NMAX];///voi tine minte lantul elementar unic de la radacina de la nodul curent
void dfs(int nod, int nivel)
{
drum[nivel]=nod;
sol[nod]=sol[drum[nivel-nrstramos[nod]]]+1;
for(int i=0; i<L[nod].size(); i++)
{
dfs(L[nod][i], nivel+1);
}
}
int main()
{
fin>>n;
for(int i=1; i<=n; i++)
fin>>nrstramos[i];
int x, y;
for(int i=1; i<=n-1; i++)
{
fin>>x>>y;///x este tatal lui y
L[x].push_back(y);
tata[y]=x;
}
///determinam radacina
int radacina=0;
for(int i=1; i<=n; i++)
if(tata[i]==0)
{
radacina=i;
break;
}
dfs(radacina, 1);
for(int i=1; i<=n; i++)
sol[i]-=1;
for(int i=1; i<=n; i++)
fout<<sol[i]<<" ";
return 0;
}