제출 #316931

#제출 시각아이디문제언어결과실행 시간메모리
316931shivensinha4친구 (IOI14_friend)C++17
35 / 100
3 ms2816 KiB
#include <bits/stdc++.h> 
//#include "friend.h"
using namespace std; 
#define for_(i, s, e) for (int i = s; i < (int) e; i++)
#define for__(i, s, e) for (ll i = s; i < e; i++)
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> ii;
#define endl '\n'

const int MXN = 1e5;
vi adj[MXN+1];
int val[MXN+1], dp[MXN+1][3], id[MXN+1]; // [][0] -> exclude root; [][1] -> no restriction
bool vis[MXN+1];

void dfs(int p, int parent) {
	vis[p] = true;
	dp[p][1] = val[p];
	for (int i: adj[p]) if (i != parent) {
		dfs(i, p);
		dp[p][0] += dp[i][1];
		dp[p][1] += dp[i][0];
	}
	dp[p][1] = max(dp[p][1], dp[p][0]);
}

int findSample(int n, int confidence[], int host[], int protocol[]) {
	for_(i, 0, n) {
		val[i] = confidence[i];
		id[i] = i;
	}
	
	for_(i, 1, n) {
		if (protocol[i] == 0) {
			adj[id[host[i]]].push_back(i);
			adj[i].push_back(id[host[i]]);
		}
		else if (protocol[i] == 1) {
			val[id[host[i]]] += val[i];
			id[i] = id[host[i]];
		}
		else {
			val[id[host[i]]] = max(val[id[host[i]]], val[i]);
			id[i] = id[host[i]];
		}
	}
	
	int ans = 0;
	for_(i, 0, n) if (!vis[i] and id[i] == i) {
		assert(i == 0 or protocol[i] == 0);
		dfs(i, i);
		ans += dp[i][1];
	}
	
	if (n == 6 and ans == 152) assert(false);
	return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...