Submission #702077

#	Time	Username	Problem	Language	Result	Execution time	Memory
702077		PCTprobability	Friend (IOI14_friend)	C++17	100 / 100	24 ms	6164 KiB

friend

#include <bits/stdc++.h>
#include "friend.h"
using namespace std;
using ll = long long;
#define pb push_back
vector<int> g[100000];
int dp[100000][2];
ll con[100000];
int findSample(int n, int confidence[], int host[], int protocol[]){
  for(int i=0;i<n;i++) con[i]=confidence[i];
  for(int i=0;i<n;i++) dp[i][1]=con[i];
  for(int i=n-1;i>=1;i--){
    dp[i][1]=max(dp[i][1],dp[i][0]);
    if(protocol[i]==0){
      dp[host[i]][0]+=dp[i][1];
      dp[host[i]][1]+=dp[i][0];
    }
    if(protocol[i]==1){
      dp[host[i]][0]+=dp[i][0];
      dp[host[i]][1]+=dp[i][1];
    }
    if(protocol[i]==2){
      dp[host[i]][1]=max(dp[host[i]][0]+dp[i][1],dp[host[i]][1]+dp[i][0]);
      dp[host[i]][0]+=dp[i][0];
    }
    dp[host[i]][1]=max(dp[host[i]][1],dp[host[i]][0]);
  }
  return max(dp[0][1],dp[0][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