Cod sursa(job #2103499)

Utilizator LauraNaduLaura Nadu LauraNadu Data 10 ianuarie 2018 13:28:07
Problema Dosare Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.83 kb
#include<fstream>
#include<vector>
#include<algorithm>
using namespace std;
ifstream f("dosare.in");
ofstream g("dosare.out");
int n, i, x, ac[16005], c[16005], d[16005];
vector<int> l[16005];
bool cmp(int a, int b)
{
    return c[a]>c[b];
}
void dfs(int nod)
{
    c[nod]=ac[nod];
    for(int i=0;i<l[nod].size();i++)
    {
        dfs(l[nod][i]);
        c[nod]+=c[l[nod][i]];
    }
    d[nod]=c[nod];
    if(l[nod].size()!=0)
    {
        sort(l[nod].begin(), l[nod].end(), cmp);
        for(int i=0;i<l[nod].size();i++)
            d[nod]+=i*c[l[nod][i]]+d[l[nod][i]];
    }
}
int main()
{
    f>>n;
    for(i=2;i<=n;i++)
    {
        f>>x;
        l[x].push_back(i);
    }
    for(i=1;i<=n;i++)
        f>>ac[i];
    dfs(1);
    /*for(i=1;i<=n;i++)
        g<<c[i]<<" ";*/
    g<<d[1];
    return 0;
}