제출 #1092113

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1092113		huyngo	친구 (IOI14_friend)	C++17	35 / 100	1 ms	604 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>  
using namespace std;
using ll = long long;
using i64 = long long;
void __print(int x) { cerr << x; } void __print(long x) { cerr << x; } void __print(long long x) { cerr << x; } void __print(unsigned x) { cerr << x; } void __print(unsigned long x) { cerr << x; } void __print(unsigned long long x) { cerr << x; } void __print(float x) { cerr << x; } void __print(double x) { cerr << x; } void __print(long double x) { cerr << x; } void __print(char x) { cerr << '\'' << x << '\''; } void __print(const char* x) { cerr << '"' << x << '"'; } void __print(const string& x) { cerr << '"' << x << '"'; } void __print(bool x) { cerr << (x ? "true" : "false"); } template<typename T, typename V> void __print(const pair<T, V>& x) { cerr << '{'; __print(x.first); cerr << ','; __print(x.second); cerr << '}'; } template<typename T> void __print(const T& x) { int f = 0; cerr << '{'; for (auto& i : x) cerr << (f++ ? "," : ""), __print(i); cerr << "}"; } void _print() { cerr << "]\n"; } template <typename T, typename... V> void _print(T t, V... v) { __print(t); if (sizeof...(v)) cerr << ", "; _print(v...); }
#define dbg(x...) cerr << "[" << #x << "] = ["; _print(x)
#define ln "\n"
#define fastIO() ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define all(x) (x).begin(), (x).end()
#define rep(i,a,b) for(int i=a; i<=b; ++i)
#define ar array
int Bit(int mask, int b) { return (mask >> b) & 1; }
const ll base = 311, MOD = 998244353, M = 1e9 + 7, INF = 1e18;

// Find out best sample
int findSample(int n, int confidence[], int host[], int protocol[]) {
	int sub = accumulate(protocol + 1, protocol + n, 0);
	if (sub == 0) { // tree
		vector<vector<int>> adj(n);
		rep(i, 1, n - 1)
			adj[host[i]].push_back(i);
		vector<vector<int64_t>> d(n, vector<int64_t>(2));
		auto dfs = [&](auto self, int u) -> void {
			vector<ar<int64_t, 2>> w;
			int64_t sum0 = 0;
			for (int v : adj[u]) {
				self(self, v);
				sum0 += d[v][0];
				w.push_back({ d[v][0], d[v][1] });
			}
			d[u][1] = confidence[u] + sum0;
			for (auto [i, j] : w)
				d[u][0] += max(i, j);
			};
		dfs(dfs, 0);
		return max(d[0][0], d[0][1]);
	}
	else if (sub == n - 1) {
		return accumulate(confidence, confidence + n, 0);
	}
	else if (sub == 2 * (n - 1)) {
		return *max_element(confidence, confidence + n);
	}
	return 0;
}

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