#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
const int N = 1e3;
vector<int> adj[N];
long long dp[N][2];
void dfs(int u, int p, int *conf){
for(int v : adj[u]){
if(v != p) dfs(v, u, conf);
}
for(int v : adj[u]){
if(v != p){
dp[u][0] += max(dp[v][1], dp[v][0]);
dp[u][1] += dp[v][0];
}
}
dp[u][1] += 1LL*conf[u];
}
int findSample(int n, int conf[],int host[],int protocol[]){
bool subtask2 = 1, subtask3 = 1, subtask4 = 1;
for(int i = 1; i < n; ++i){
if(protocol[i] != 1) subtask2 = 0;
if(protocol[i] != 2) subtask3 = 0;
if(protocol[i] != 0) subtask4 = 0;
}
if(subtask2){
long long ans = 0;
for(int i = 0; i < n; ++i) ans += 1LL*conf[i];
return ans;
}
if(subtask3){
return *max_element(conf, conf+n);
}
if(subtask4){
//build tree
for(int i = 1; i < n; ++i){
adj[host[i]].push_back(i);
adj[i].push_back(host[i]);
}
dfs(0, 0, conf);
return max(dp[0][1], dp[0][0]);
}
return -1;
}
