#include "friend.h"
#include <vector>
// meow meow meow mewo mrrow nya nya :3c
// go play vivid/stasis! it's very awesome gmae on steam
using namespace std;
vector<int> friends[10000];
int dfs(int i, int l, bool d, int confidence[]) {
int res = confidence[i] * d;
for (int j : friends[i]) {
if (j == l) continue;
res += dfs(j, i, !d, confidence);
}
return res;
}
int findSample(int n, int confidence[], int host[], int protocol[]) {
for (int i = 1; i < n; i++) friends[host[i]].push_back(i), friends[i].push_back(host[i]);
int res1 = dfs(0, -1, false, confidence);
int res2 = dfs(0, -1, true, confidence);
return res1 > res2 ? res1 : res2;
}
