Submission #922046

#	Time	Username	Problem	Language	Result	Execution time	Memory
922046		Macker	Friend (IOI14_friend)	C++14	100 / 100	21 ms	3416 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
#define all(v) v.begin(), v.end()



// Find out best sample
int findSample(int n,int confidence[],int host[],int protocol[]){
	vector<int> c(n), nc(n, 0);

	for (int i = 0; i < n; i++) {
		c[i] = confidence[i];
	}
	

	for (int i = n - 1; i > 0; i--) {
		if(protocol[i] == 0){
			c[host[i]] = nc[i] + c[host[i]];
			nc[host[i]] = max(nc[host[i]] + nc[i], nc[host[i]] + c[i]);
		}
		else if(protocol[i] == 1){
			c[host[i]] = max(nc[host[i]] + c[i], max(c[host[i]] + nc[i], c[host[i]] + c[i]));
			nc[host[i]] = nc[i] + nc[host[i]];
		}
		else {
			c[host[i]] = max(c[host[i]] + nc[i], c[i] + nc[host[i]]);
			nc[host[i]] = nc[i] + nc[host[i]];
		}
	}
	return max(c[0], nc[0]);
}

• Subtask 1: 11 / 11
• Subtask 2: 8 / 8
• Subtask 3: 8 / 8
• Subtask 4: 19 / 19
• Subtask 5: 23 / 23
• Subtask 6: 31 / 31