Pagini recente » Cod sursa (job #1634171) | Cod sursa (job #539413) | Cod sursa (job #347272) | Cod sursa (job #772403) | Cod sursa (job #1390578)
#include <algorithm>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("dosare.in");
ofstream fout("dosare.out");
typedef long long i64;
const int nmax= 16000;
i64 sol= 0, v[nmax+1];
vector <int> g[nmax+1];
void dfs( int x ) {
vector <i64> o;
for ( vector <int>::iterator it= g[x].begin(); it!=g[x].end(); ++it ) {
dfs(*it);
v[x]= (i64)v[x]+v[*it];
o.push_back(v[*it]);
}
sort( o.begin(), o.end() ) ;
sol= (i64)sol+v[x];
for ( int i= (int)o.size()-1; i>=0; --i ) {
sol= (i64)sol+(i64)((int)o.size()-1-i)*o[i];
}
}
int main( ) {
int n;
fin>>n;
for ( int i= 2; i<=n; ++i ) {
int x;
fin>>x;
g[x].push_back(i);
}
for ( int i= 1; i<=n; ++i ) {
fin>>v[i];
}
dfs(1);
fout<<sol<<"\n";
return 0;
}