Submission #832201

#	Time	Username	Problem	Language	Result	Execution time	Memory
832201		tranxuanbach	Split the Attractions (IOI19_split)	C++17	18 / 100	60 ms	12608 KiB

split

#include "split.h"

#include <bits/stdc++.h>
using namespace std;

#define isz(a) ((signed)a.size())

const int N = 1e5 + 5, M = 2e5 + 5;

int n, m;
int a, b, c;
vector <int> adj[N];

vector <int> ans;

namespace subtask1{
	bool check(){
		for (int u = 0; u < n; u++){
			if (isz(adj[u]) > 2){
				return false;
			}
		}
		return true;
	}

	vector <bool> chosen;

	void dfs(int u){
		chosen[u] = true;
		if (a){
			ans[u] = 1;
			a--;
		}
		else if (b){
			ans[u] = 2;
			b--;
		}
		else{
			ans[u] = 3;
			c--;
		}
		for (auto v: adj[u]){
			if (chosen[v]){
				continue;
			}
			dfs(v);
		}
	}

	vector <int> solve(){
		chosen.assign(n, false);

		int s = 0;
		for (int u = 1; u < n; u++){
			if (isz(adj[u]) == 1){
				s = u;
				break;
			}
		}
		dfs(s);
		return ans;
	}
}

vector <int> find_split(int _n, int _a, int _b, int _c, vector <int> _u, vector <int> _v){
	n = _n; m = isz(_u); a = _a; b = _b; c = _c;
	for (int i = 0; i < m; i++){
		int u = _u[i], v = _v[i];
		adj[u].emplace_back(v);
		adj[v].emplace_back(u);
	}
	ans.assign(n, 0);

	if (subtask1::check()){
		return subtask1::solve();
	}
}

Compilation message (stderr)

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:77:1: warning: control reaches end of non-void function [-Wreturn-type]
   77 | }
      | ^

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