제출 #958063

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

friend

#include "friend.h"
#include <bits/stdc++.h>

using namespace std;

#define scd(t) scanf("%d", &t)
#define sclld(t) scanf("%lld", &t)
#define forr(i, j, k) for (int i = j; i < k; i++)
#define frange(i, j) forr(i, 0, j)
#define all(cont) cont.begin(), cont.end()
#define mp make_pair
#define pb push_back
#define f first
#define s second
typedef long long int lli;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<bool> vb;
typedef vector<lli> vll;
typedef vector<string> vs;
typedef vector<pii> vii;
typedef vector<vi> vvi;
typedef map<int, int> mpii;
typedef set<int> seti;
typedef multiset<int> mseti;
typedef long double ld;

vvi graph;
vi val;

pii dfs(int x, int p) {
	int v1 = val[x];
	int v2 = 0;

	for(auto e : graph[x]) {
		if(e != p) {
			auto out = dfs(e, x);
			v1 += out.s;
			v2 += max(out.f, out.s);
		}
	}
	return mp(v1, v2);
}

vi pos;

int findf(int x) {
	if(x == pos[x]) return x;
	pos[x] = findf(pos[x]);
	return pos[x];
}

int findSample(int n,int confidence[],int host[],int protocol[]){
	graph = vvi(n);
	val = vi(n);

	val[0] = confidence[0];

	pos = vi(n);
	pos[0] = 0;

	forr(i, 1, n) {
		if(protocol[i] == 0) {
			val[i] = confidence[i];
			graph[findf(host[i])].pb(i);
			graph[i].pb(findf(host[i]));
			pos[i] = i;
		}
		else if(protocol[i] == 1) {
			val[findf(host[i])] += confidence[i];
			pos[i] = findf(host[i]);
		}
		else {
			val[findf(host[i])] = max(val[findf(host[i])], confidence[i]);
			pos[i] = findf(host[i]);
		}
	}

	auto out = dfs(0, -1);
	return max(out.f, out.s);
}

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