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