Pagini recente » Cod sursa (job #867697) | Cod sursa (job #1108371) | Cod sursa (job #2962302) | Cod sursa (job #2562257) | Cod sursa (job #2102148)
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
ifstream fin ("dosare.in");
ofstream fout ("dosare.out");
int n,i,x,ac[16010],D[16010],sons[16010];
vector <int> L[16001];
int cmp (int a, int b){
return sons[a]>sons[b];
}
void dfs (int nod){
D[nod]=0;
sons[nod]=ac[nod];
for(int i=0;i<L[nod].size();i++){
dfs(L[nod][i]);
sons[nod]+=sons[L[nod][i]];
}
//daca nod are fii, il sortam descrescator dupa sons
//in dinamica tinem nr total de apasari daca nod ar fi radacina
int c=L[nod].size();
if(c){
sort(L[nod].begin(), L[nod].end(), cmp);
for(int i=0;i<L[nod].size();i++)
D[nod]+=i*sons[L[nod][i]]+D[L[nod][i]];
}
D[nod]+=sons[nod];
}
int main(){
fin>>n;
for(i=2;i<=n;i++){
fin>>x;
L[x].push_back(i);
}
for(i=1;i<=n;i++)
fin>>ac[i];
dfs(1);
fout<<D[1];
return 0;
}