Submission #797314

#	Time	Username	Problem	Language	Result	Execution time	Memory
797314		Sohsoh84	Friend (IOI14_friend)	C++17	100 / 100	24 ms	3404 KiB

friend

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

using namespace std;

#define debug(x)		cerr << #x << ": " << x << endl;

const int MAXN = 1e5 + 10;

int C[MAXN], P[MAXN], Q[MAXN];

int findSample(int n, int C_[], int host[], int protocol[]){
	for (int i = 0; i < n; i++)
		P[i] = C_[i];

	for (int i = n - 1; i > 0; i--) {
		int h = host[i];
		if (protocol[i] == 2) {
			P[h] = max(P[h] + Q[i], P[i] + Q[h]);
			Q[h] = Q[i] + Q[h];
		} else if (protocol[i] == 1) {
			P[h] = max({P[h] + P[i], P[h] + Q[i], P[i] + Q[h]});
			Q[h] = Q[i] + Q[h];
		} else {
			P[h] = P[h] + Q[i];
			Q[h] = Q[h] + max(P[i], Q[i]);
		}

		P[h] = max(P[h], Q[h]);
	}

	return max(Q[0], P[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