Submission #603455

#	Time	Username	Problem	Language	Result	Execution time	Memory
603455		SeDunion	Split the Attractions (IOI19_split)	C++17	0 / 100	4 ms	5008 KiB

split

#include "split.h"
#include<iostream>
#include<vector>

using namespace std;
const int N = 2e5 + 123;

vector<int>g[N];

int A, B;

int used[N];

vector<int>vec;

void dfs(int v) {
	vec.emplace_back(v);
	used[v] = 1;
	for (int to : g[v]) if (!used[to]) {
		dfs(to);
	}
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
	vector<int>ans(n);
	for (int i = 0 ; i < n - 1 ; ++ i) {
		g[p[i]].emplace_back(q[i]);
		g[q[i]].emplace_back(p[i]);
	}
	if (a > b) swap(a, b);
	if (b > c) swap(b, c);
	if (a > b) swap(a, b);
	A = a, B = b;
	dfs(0);
	for (int i = 0 ; i < a ; ++ i) ans[vec[i]] = 1;
	for (int i = 0 ; i < b ; ++ i) ans[vec[i+a]] = 2;
	for (int i = 0 ; i < c ; ++ i) ans[vec[i+a+b]] = 3;
	return ans;
}

• Subtask 1: 0 / 7
• Subtask 2: 0 / 11
• Subtask 3: 0 / 22
• Subtask 4: 0 / 24
• Subtask 5: 0 / 36