Submission #586590

#	Time	Username	Problem	Language	Result	Execution time	Memory
586590		LastRonin	친구 (IOI14_friend)	C++14	0 / 100	1 ms	212 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
using namespace std;

const ll N = 2e3 + 10;

ll p[N], q[N];

int findSample(int n,int confidence[],int host[],int protocol[]){
	for(int i = n - 1; i > 0; i--) {
		if(protocol[i] == 0) {
			// 1
			p[host[i]] = p[host[i]] + q[i];
			q[host[i]] = max(q[host[i]] + p[i], q[host[i]] + q[i]);
		} else if(protocol[i] == 1) {
			// withfriends
			p[host[i]] = max({p[host[i]] + q[i], p[host[i]] + p[i], q[host[i]] + p[i]});
			q[host[i]] = q[host[i]] + q[i];
		} else {
			p[host[i]] = max({p[host[i]] + q[i], q[host[i]] + p[i]});
			q[host[i]] = q[host[i]] + q[i];
		}
	}
	return max(p[0], q[0]);
}
/*
6
13 3 6 20 10 15
0 0 0 1 1 2 2 1 0 0
6
13 3 6 20 10 15
0 0 0 1 1 2 2 1 0 0

*/
/*
#define __MAXSIZE__ 100002
#include "friend.h"
using namespace std;

// Confidence
int confidence[__MAXSIZE__];

// Host
int host[__MAXSIZE__];

// Protocol
int protocol[__MAXSIZE__];

// Main
int main(void)
{
	int n,i;

	// Number of people
	assert(scanf("%d",&n)==1);
	
	// Confidence
	for(i=0;i<n;i++)
		assert(scanf("%d",&confidence[i])==1);
	
	// Host and Protocol
	for(i=1;i<n;i++)
		assert(scanf("%d %d",&host[i],&protocol[i])==2);
	
	// Answer
	printf("%d\n",findSample(n,confidence,host,protocol));
	return 0;
}
/**/

Compilation message (stderr)

friend.cpp:71:1: warning: "/*" within comment [-Wcomment]
   71 | /**/
      |

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