Submission #932795

#	Time	Username	Problem	Language	Result	Execution time	Memory
932795		vjudge1	Friend (IOI14_friend)	C++17	27 / 100	2 ms	4548 KiB

friend

#include<bits/stdc++.h>
 
#pragma GCC optimize("Ofast")
 
using namespace std;
 
#define F first
#define S second
#define ll long long
// #define int ll
#define pb push_back
#define sz(s) (int)s.size()
#define pii pair<int,int>
#define all(v) v.begin(),v.end()
#define mem(a,i) memset(a,i,sizeof(a))
#define in insert
#define lb lower_bound
#define ub upper_bound
#define y1 yy
#define ppb pop_back
#define ull unsigned ll
 
const int MAX=1e5+55;
const int inf=1e9;
const int N=2e5;
const int C=331;
const int C1=431;
const int mod=1e9+9;
const int mod1=1e9+9;
// #include "friend.h"

int dp[MAX][2];
vector<pii> g[MAX];
int a[MAX];

void dfs(int v,int p=-1){
  for(auto to:g[v]){
    if(to.F==p)continue;
    dfs(to.F,v);
    dp[v][0]+=max(dp[to.F][0],dp[to.F][1]);
    if(to.S==0)dp[v][1]+=dp[to.F][0];
    else dp[v][1]+=max(dp[to.F][0],dp[to.F][1]);
  }
  dp[v][1]+=a[v];
}

// Find out best sample
int findSample(int n,int confidence[],int host[],int protocol[]){
  for(int i=0;i<n;i++)a[i]=confidence[i];
	for(int i=1;i<n;i++){
    if(protocol[i]==0){
      g[i].pb({host[i],0});
      g[host[i]].pb({i,0});
    }
    else{
      g[host[i]].pb({i,1});
      g[i].pb({host[i],1});
    }
  }
  dfs(0);
  return max(dp[0][0],dp[0][1]);
}

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