Cod sursa(job #2088035)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 14 decembrie 2017 18:20:13
Problema Dosare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.76 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];
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;
}