#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
#define FOR(i, a, b) for (int i = (a); i <= (b); i++)
#define REP(n) FOR(O, 1, (n))
#define f first
#define s second
#define pb push_back
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<pii> vii;
const int MAXN = 100100;
int n;
int conf[MAXN], h[MAXN], prot[MAXN];
int sub2 () {
int ret = 0;
FOR(i, 0, n-1) ret += conf[i];
return ret;
}
int sub3 () {
int ret = 0;
FOR(i, 0, n-1) ret = max(ret, conf[i]);
return ret;
}
int dp[MAXN][2];
vi adj[MAXN];
void dfs (int v, int p) {
dp[v][1] = conf[v];
for (int u : adj[v]) {
if (u == p) continue;
dfs (u, v);
dp[v][0] += max(dp[u][0], dp[u][1]);
dp[v][1] += dp[u][0];
}
}
int sub4 () {
FOR(i, 1, n-1) adj[h[i]].pb(i);
dfs (0, -1);
return max(dp[0][1], dp[0][0]);
}
int findSample(int N,int confidence[],int host[],int protocol[]){
n = N;
FOR(i, 0, n-1) conf[i] = confidence[i];
FOR(i, 1, n-1) h[i] = host[i], prot[i] = protocol[i];
bool is2 = true;
bool is3 = true;
bool is4 = true;
FOR(i, 1, n-1) {
if (protocol[i] != 1) is2 = false;
if (protocol[i] != 2) is3 = false;
if (protocol[i] != 0) is4 = false;
}
if (is2)
return sub2();
else if (is3)
return sub3();
else if (is4)
return sub4();
else
return sub4();
}
