Pagini recente » Cod sursa (job #2495089) | Cod sursa (job #456780) | Cod sursa (job #203787) | Cod sursa (job #1491106) | Cod sursa (job #639966)
Cod sursa(job #639966)
# include <cstdio>
# include <set>
# include <stack>
# include <vector>
using namespace std;
# define y second
const char *FIN = "guvern.in", *FOU = "guvern.out";
const int MAX = 100005, oo = 0x3f3f3f3f;
bool viz[MAX];
int N, dp[MAX], grad[MAX];
vector <int> G[MAX], T[MAX][2];
stack < int, vector <int> > son[MAX];
set < pair <int, int> > S;
int search (int nod, int c) {
int sol = 0;
for (vector <int> :: iterator it = T[nod][c].begin (); it != T[nod][c].end (); ++it)
sol += search (*it, 0);
return max (dp[nod], sol);
}
void dfs (int nod) {
viz[nod] = 1;
set < pair <int, int> > :: iterator it = S.upper_bound (make_pair (grad[nod], 0));
if (son[it -> y].empty ())
T[it -> y][1].push_back (nod);
else T[son[it -> y].top ()][0].push_back (nod);
son[it -> y].push (nod), S.insert (make_pair (grad[nod], nod));
for (vector <int> :: iterator i = G[nod].begin (); i != G[nod].end (); ++i) {
if (viz[*i]) continue;
dfs (*i);
}
S.erase (make_pair (grad[nod], nod)), son[it -> y].pop ();
dp[nod] = search (nod, 1) + 1;
}
void init (void) {
grad[0] = oo, G[0].push_back (1), G[1].push_back (0);
}
int main (void) {
freopen (FIN, "r", stdin);
scanf ("%d", &N);
for (int i = 1, x, y; i < N; ++i) {
scanf ("%d %d", &x, &y);
G[x].push_back (y);
G[y].push_back (x);
}
for (int i = 1; i <= N; ++i)
scanf ("%d", grad + i);
init (), dfs (0);
fprintf (fopen (FOU, "w"), "%d", --dp[0]);
}