Cod sursa(job #639966)

Utilizator cont_de_testeCont Teste cont_de_teste Data 24 noiembrie 2011 14:20:47
Problema Guvern Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.55 kb
# 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]);
}