Cod sursa(job #2088019)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 14 decembrie 2017 18:05:16
Problema Dosare Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.68 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];
vector<int>v[16002];
bool cmp(int a, int b)
{
    return sum[a]<sum[b];
}
void dfs(int nod)
{
    sum[nod]+=Hmm[nod];
    for(int i=0;i<v[nod].size();++i)
        dfs(v[nod][i]);
    sort(v[nod].begin(),v[nod].end(),cmp);
    for(long long i=0;i<v[nod].size();++i)
        sum[nod]+=sum[v[nod][i]]*(i+1);
}
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;
}