#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
void debug_out() {cerr<<endl;}
template <typename Head, typename... Tail>
void debug_out(Head _H, Tail... _T) {cerr<<" "<<to_string(_H);debug_out(_T...);}
#define debug(...) cerr<<"["<<#__VA_ARGS__<<"]:",debug_out(__VA_ARGS__)
const int MAXN = -1;
const int inf=1000000500ll;
const int MOD = (int)1e9 + 7;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
typedef pair<int,int> pi;
// Find out best sample
int findSample(int n,int confidence[],int host[],int protocol[]){
int oth[n+5];memset(oth,0,sizeof oth);
for(int i=n-1;i>=0;i--){
if(i==0)break;
if(protocol[i] == 0){
confidence[host[i]]=confidence[host[i]]+oth[i];
oth[host[i]] += max(confidence[i], oth[i]);
} else if(protocol[i] == 1){
confidence[host[i]] += max(confidence[i], oth[i]);
oth[host[i]]=oth[host[i]]+oth[i];
} else{
confidence[host[i]]=max(confidence[i],confidence[host[i]]+oth[i]);
oth[host[i]]=oth[host[i]]+oth[i];
}
}
return max(confidence[0],oth[0]);
}
