Cod sursa(job #586699)

Utilizator GavrilaVladGavrila Vlad GavrilaVlad Data 2 mai 2011 19:51:53
Problema Guvern Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.92 kb
#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, 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], make_pair(-niv[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], make_pair(-niv[nod], nod)));

    set<pair<int, pair<int, int> > > ::iterator it=g.lower_bound(make_pair(coop[nod], make_pair(-maxn, 0)));
    if(it!=g.end())
        prov[(it->second).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;
}