Submission #484948

#	Time	Username	Problem	Language	Result	Execution time	Memory
484948		couplefire	Friend (IOI14_friend)	C++17	100 / 100	25 ms	3352 KiB

friend

#include <bits/stdc++.h>
#include "friend.h"
 
using namespace std;
 
const int N = (int)1e5 + 7;
 
int dp[N][2];
 
int findSample(int n, int confidence[], int host[], int protocol[]){
  for (int i = 0; i < n; i++) 
  	dp[i][0] = confidence[i];
  for (int i = n - 1; i > 0; i--) {
  	if (protocol[i] == 0) {
  		dp[host[i]][0] += dp[i][1];
  		dp[host[i]][1] += dp[i][0];
  	} else if (protocol[i] == 1) {
  		dp[host[i]][0] += dp[i][0];
  		dp[host[i]][1] += dp[i][1];
  	} else if (protocol[i] == 2) {
  		dp[host[i]][0] = max(dp[host[i]][0] + dp[i][1], dp[host[i]][1] + dp[i][0]);
  		dp[host[i]][1] += dp[i][1];
  	}
  	dp[host[i]][0] = max(dp[host[i]][0], dp[host[i]][1]);           
  }
  return max(dp[0][0], dp[1][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