#include "friend.h"
#include<bits/stdc++.h>
using namespace std;
int findSample(int n, int confidence[], int host[], int protocol[]){
vector<int> a(n), b(n, 0);
for (int i=0; i<n; i++) a[i] = confidence[i];
for (int i=n-1; i>=1; i--){
if (protocol[i] == 0){
a[host[i]] += b[i];
b[host[i]] += max(a[i], b[i]);
}
if (protocol[i] == 1){
a[host[i]] = max(a[host[i]]+max(a[i], b[i]), a[i]+b[host[i]]);
b[host[i]] += b[i];
}
if (protocol[i] == 2){
a[host[i]] = max(a[host[i]]+b[i], a[i]+b[host[i]]);
b[host[i]] += b[i];
}
}
return max(a[0], b[0]);
}
