Pagini recente » Cod sursa (job #1301346) | Cod sursa (job #2255598) | Cod sursa (job #2909897) | Cod sursa (job #772144) | Cod sursa (job #1501067)
#include <fstream>
#include <vector>
#include <string.h>
#define nMax 100002
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int Lev[nMax], K[nMax], a, b, Sol[nMax], i, n, root, m, nr, poz;
vector <int> G[nMax];
bool Deg[nMax];
char sir[800000];
void DFS(int node, int lvl)
{
if(K[node]==0)
Lev[lvl]=0;
else
Lev[lvl]=Lev[lvl-K[node]]+1;
Sol[node]=Lev[lvl];
for(vector<int>::iterator it=G[node].begin();it!=G[node].end();it++)
DFS(*it, lvl+1);
}
int main()
{
fin>>n;
fin.get();
fin.getline(sir+1, 800000);
m=strlen(sir+1);
for(i=1;i<=m;i++)
{
if(sir[i]>='0'&&sir[i]<='9')
nr=nr*10+sir[i]-'0';
else
{
K[++poz]=nr;
nr=0;
}
}
for(i=1;i<n;i++)
{
fin>>a>>b;
G[a].push_back(b);
Deg[b]=1;
}
for(i=1;i<=n;i++)
{
if(Deg[i]==0)
{
root=i;
break;
}
}
DFS(root, 1);
for(i=1;i<=n;i++)
fout<<Sol[i]<<" ";
return 0;
}