Cod sursa(job #457341)

Utilizator DraStiKDragos Oprica DraStiK Data 18 mai 2010 23:07:06
Problema Dosare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.97 kb
#include <algorithm>
#include <vector>
using namespace std;

#define pb push_back
#define DIM 16005

vector <long long> fii;
vector <int> g[DIM];
long long cnt[DIM];
long long nrt;
int n;

void read ()
{
    int i,x;

    scanf ("%d",&n);
    for (i=2; i<=n; ++i)
    {
        scanf ("%d",&x);
        g[x].pb (i);
    }
    for (i=1; i<=n; ++i)
        scanf ("%d",&cnt[i]);
}

void df (int nod)
{
    vector <int> :: iterator it;
    int i;

    for (it=g[nod].begin (); it!=g[nod].end (); ++it)
        df (*it);
    fii.clear ();
    for (it=g[nod].begin (); it!=g[nod].end (); ++it)
    {
        cnt[nod]+=cnt[*it];
        fii.pb (cnt[*it]);
    }
    nrt+=cnt[nod];
    sort (fii.rbegin (),fii.rend ());
    for (i=0; i<(int)fii.size (); ++i)
        nrt+=i*fii[i];
}

int main ()
{
    freopen ("dosare.in","r",stdin);
    freopen ("dosare.out","w",stdout);

    read ();
    df (1);
    printf ("%lld",nrt);

    return 0;
}