Submission #719348

#	Time	Username	Problem	Language	Result	Execution time	Memory
719348		Ahmed57	Friend (IOI14_friend)	C++14	100 / 100	26 ms	3136 KiB

friend

#include <bits/stdc++.h>
using namespace std;

const int mxN=1e5;
int dp[mxN][2];

int findSample(int n, int c[], int h[], int p[]) {
	for(int i=0; i<n; ++i)
		dp[i][1]=c[i];
	for(int i=n-1; i; --i) {
		if(p[i]==0) {
			dp[h[i]][0]+=dp[i][1];
			dp[h[i]][1]+=dp[i][0];
		} else if(p[i]==1) {
			dp[h[i]][0]+=dp[i][0];
			dp[h[i]][1]+=dp[i][1];
		} else {
			dp[h[i]][1]=max(dp[h[i]][0]+dp[i][1], dp[h[i]][1]+dp[i][0]);
			dp[h[i]][0]+=dp[i][0];
		}
		dp[h[i]][1]=max(dp[h[i]][0], dp[h[i]][1]);
	}
	return max(dp[0][0], dp[0][1]);
}

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