Submission #554264

#	Time	Username	Problem	Language	Result	Execution time	Memory
554264		flappybird	Friend (IOI14_friend)	C++17	100 / 100	32 ms	3452 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> pii;

#define MAX 101010

pii dp[MAX];

int findSample(int N, int confidence[], int host[], int protocol[]) {
	int i;
	for (i = 0; i < N; i++) dp[i].first = confidence[i];
	for (i = N - 1; i > 0; i--) {
		if (protocol[i] == 0) {
			dp[host[i]].first += dp[i].second;
			dp[host[i]].second += max(dp[i].first, dp[i].second);
		}
		else if (protocol[i] == 1) {
			dp[host[i]].first = max(dp[host[i]].first + max(dp[i].first, dp[i].second), dp[host[i]].second + dp[i].first);
			dp[host[i]].second += dp[i].second;
		}
		else {
			dp[host[i]].first = max(dp[host[i]].first + dp[i].second, dp[host[i]].second + dp[i].first);
			dp[host[i]].second += dp[i].second;
		}
	}
	return max(dp[0].first, dp[0].second);
}

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