Pagini recente » Cod sursa (job #1064412) | Cod sursa (job #3255345) | Cod sursa (job #2215661) | Cod sursa (job #143900) | Cod sursa (job #586695)
Cod sursa(job #586695)
#include <stdio.h>
#include <vector>
#include <set>
using namespace std;
#define maxn 200010
#define mlog 20
int n, i, j, k, a, b, el, rez, coop[maxn], f[maxn], niv[maxn], d[maxn], sv[maxn], dsv[maxn];
vector<int> v[maxn], prov[maxn];
set<pair<int, int> > g;
int st[mlog][maxn];
int verif(int nc, int nod)
{
for(int j=mlog-1; j>=0; --j)
{
if(st[j][nc]==0)
continue;
if(niv[st[j][nc]]>=niv[nod])
nc=st[j][nc];
}
return (nc==nod);
}
void df(int nod, int tata)
{
if(f[nod]==1)
return;
f[nod]=1;
niv[nod]=niv[tata]+1;
st[0][nod]=tata;
for(int i=1; st[i-1][st[i-1][nod]]>0; ++i)
st[i][nod]=st[i-1][st[i-1][nod]];
g.insert(make_pair(coop[nod], nod));
for(int i=0; i<v[nod].size(); ++i)
if(f[v[nod][i]]==0)
df(v[nod][i], nod);
g.erase(make_pair(coop[nod], nod));
set<pair<int, int> > ::iterator it=g.lower_bound(make_pair(coop[nod], 0));
if(it!=g.end())
prov[it->second].push_back(nod);
sv[0]=0;
int sum=0;
for(int i=0; i<prov[nod].size(); ++i)
{
sum=0;
while(sv[0]>0 && verif(sv[sv[0]], prov[nod][i]))
sum+=dsv[sv[0]--];
sv[++sv[0]]=prov[nod][i];
dsv[sv[0]]=max(d[prov[nod][i]], sum);
}
// printf("%d\n", nod);
for(int i=1; i<=sv[0]; ++i)
{
// printf("%d %d\n", sv[i], dsv[i]);
d[nod]+=dsv[i];
}
++d[nod];
// printf("%d\n\n", d[nod]);
rez=max(rez, d[nod]);
}
int main()
{
freopen("guvern.in", "r", stdin);
freopen("guvern.out", "w", stdout);
scanf("%d", &n);
for(int i=1; i<n; ++i)
{
scanf("%d%d", &a, &b);
v[a].push_back(b);
v[b].push_back(a);
}
for(int i=1; i<=n; ++i)
scanf("%d", &coop[i]);
df(1, 0);
printf("%d\n", rez);
return 0;
}