Submission #1068128

#	Time	Username	Problem	Language	Result	Execution time	Memory
1068128		LittleOrange	Split the Attractions (IOI19_split)	C++17	7 / 100	38 ms	9948 KiB

split

#include "split.h"

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

vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
	vector<vector<ll>> con(n);
	ll m = p.size();
	for(ll i = 0;i<m;i++){
		con[p[i]].push_back(q[i]);
		con[q[i]].push_back(p[i]);
	}
	ll mxd = 0;
	for(auto &o : con) mxd = max(mxd,(ll)o.size());
	if (mxd==2){
		ll st = 0;
		for(ll i = 0;i<n;i++) if(con[i].size()==1) st = i;
		vector<ll> v(1,st);
		vector<ll> u(n,0);
		u[st] = 1;
		while(v.size()<n){
			for(ll i : con[st]) if(!u[i]){
				u[i] = 1;
				v.push_back(i);
				st = i;
				break;
			}
		}
		vector<ll> ret(n,0);
		for(ll i = 0;i<a;i++) ret[v[i]] = 1;
		for(ll i = a;i<a+b;i++) ret[v[i]] = 2;
		for(ll i = a+b;i<a+b+c;i++) ret[v[i]] = 3;
		return ret;
	}
	if (a==1){
		vector<ll> r(n,0);
		function<void(ll)> dfs;
		dfs = [&](ll i){
			if (r[i]||b==0) return;
			r[i] = 2;
			b--;
			for(ll j : con[i]){
				dfs(j);
			}
		};
		for(ll i = 0;i<n;i++)if(!r[i]){
			if (a){
				a=0;
				r[i] = 1;
			}else r[i] = 3;
		}
		dfs(0);
		return r;
	}
	vector<int> res;
	if (n == 9) {
		res = {1, 1, 3, 1, 2, 2, 3, 1, 3};
	} else {
		res = {0, 0, 0, 0, 0, 0};
	}
	return res;
}

Compilation message (stderr)

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:22:17: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   22 |   while(v.size()<n){
      |         ~~~~~~~~^~

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