Cod sursa(job #116815)

Utilizator wefgefAndrei Grigorean wefgef Data 19 decembrie 2007 17:26:33
Problema Dosare Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;

#define pb push_back
#define sz(c) int((c).size())
#define rall(c) (c).rbegin(), (c).rend()
#define tr(it, c) for (typeof((c).begin()) it = (c).begin(); it != (c).end(); ++it)

const int Nmax = 16005;

int N;
int Press[Nmax];
vector<int> G[Nmax];
long long Ret;

void ReadData() {
	scanf("%d", &N);
	for (int i = 2; i <= N; ++i) {
		int father;
		scanf("%d", &father);
		G[father].pb(i);
	}
	for (int i = 1; i <= N; ++i)
		scanf("%d", Press+i);
}

void DF(int k) {
	vector<int> T;
	tr(it, G[k]) {
		DF(*it);
		Press[k] += Press[*it];
		T.pb(Press[*it]);
	}
	sort(rall(T));
	for (int i = 0; i < sz(T); ++i)
		Ret += (long long)(i+1) * T[i];
}

void WriteData() {
	printf("%lld\n", Ret + Press[1]);
}

int main() {
	freopen("dosare.in", "r", stdin);
	freopen("dosare.out", "w", stdout);

	ReadData();
	DF(1);
	WriteData();
}