#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+100;
int sm=0,smp=0,tp[N];
int findSample(int n,int confidence[],int host[],int protocol[]){
set<int> cur;
for(int i=1;i<n;i++)cur.insert(protocol[i]);
if(cur.size()==1)
{
if((*begin(cur))==0)
{
tp[0]=0;
sm=smp=confidence[0];
for(int i=1;i<n;i++)
{
sm+=confidence[i];
tp[i]=1-tp[host[i]];
if(!tp[i])smp+=confidence[i];
}
return max(sm,sm-smp);
}
else if((*begin(cur))==1)
{
sm=confidence[0];
for(int i=0;i<n;i++)sm=max(sm,confidence[i]);
return sm;
}
else
{
sm=0;
for(int i=0;i<n;i++)
{
sm+=confidence[i];
}
return sm;
}
}
int ans=10;
return ans;
}
