Submission #832219

#	Time	Username	Problem	Language	Result	Execution time	Memory
832219		tranxuanbach	Split the Attractions (IOI19_split)	C++17	18 / 100	60 ms	12664 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;
vector <pair <int, int>> vcolor;
vector <int> adj[N];

vector <int> ans;

namespace subtask12{
	bool check(){
		for (auto &[cnt, col]: vcolor){
			if (col == 1 and cnt == 1){
				return true;
			}
		}
		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;
		ans[u] = vcolor.back().second;
		vcolor.back().first--;
		if (vcolor.back().first == 0){
			vcolor.pop_back();
		}
		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);
	vcolor = {{a, 1}, {b, 2}, {c, 3}};
	sort(vcolor.begin(), vcolor.end());
	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 (subtask12::check()){
		return subtask12::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