Cod sursa(job #2088034)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 14 decembrie 2017 18:17:03
Problema Dosare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.81 kb
#include<bits/stdc++.h>
using namespace std;
ifstream f("dosare.in");
ofstream g("dosare.out");
long long n,Hmm[16002];
long long sum[16002],a[16002];
vector<int>v[16002];
bool cmp(int a, int b)
{
    return sum[a]<sum[b];
}
void dfs(int nod)
{
    int Len=v[nod].size();
    for(int i=0;i<Len;++i)
        dfs(v[nod][i]);
    for(int i=0;i<Len;i++)
    {
        sum[nod]+=sum[v[nod][i]];
        a[i]=Hmm[v[nod][i]];
        Hmm[nod]+=Hmm[v[nod][i]];
    }
    sort(a,a+Len,greater<int>());
    for (int i=Len-1; i>=0; i--)
        sum[nod]+=a[i]*i;
    sum[nod]+=Hmm[nod];
}
int main()
{
    f>>n;
    for(int i=2;i<=n;++i)
    {
        int nr;
        f>>nr;
        v[nr].push_back(i);
    }
    for(int i=1;i<=n;++i)
        f>>Hmm[i];
    dfs(1);
    g<<sum[1]<<'\n';
    return 0;
}