Submission #1014788

#	Time	Username	Problem	Language	Result	Execution time	Memory
1014788		alex_2008	Split the Attractions (IOI19_split)	C++14	0 / 100	1047 ms	1048576 KiB

split

#include "split.h"
#include <iostream>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10, M = 110;
vector <vector<int>> G;
vector<int> find_split(int n, int A, int B, int C, vector<int> p, vector<int> q) {
	int m = (int)p.size();
	bool sb1 = true, sb2 = true;
	G.resize(n);
	for (int i = 0; i < m; i++) {
		G[p[i]].push_back(q[i]);
		G[q[i]].push_back(p[i]);
	}
	for (int i = 0; i < n; i++) {
		if ((int)G[i].size() > 2) sb1 = false;
	}
	if (sb1) {
		int a = -1, b = -1;
		for (int i = 0; i < n; i++) {
			if ((int)G[i].size() == 1) {
				if (a == -1) a = i;
				else b = i;
			}
		}
		if (a != -1) {
			vector <int> v;
			v.push_back(a);
			v.push_back(G[a][0]);
			while (v.back() != b) {
				int a = v[(int)v.size() - 1];
				int b = v[(int)v.size() - 2];
				if (G[a][0] == b) v.push_back(G[a][1]);
				else v.push_back(G[a][0]);
			}
		}
		vector <int> v;
		v.push_back(a);
		v.push_back(G[a][0]);
		while (v.size() != n) {
			int last = v.back();
			if (G[last][0] != v[(int)v.size() - 2]) v.push_back(G[last][0]);
			else v.push_back(G[last][1]);
		}
		vector <int> ans(n);
		for (int i = 0; i < n; i++) {
			if (i < A) ans[v[i]] = 1;
			else if (i < A + B) ans[v[i]] = 2;
			else ans[v[i]] = 3;
		}
		return ans;
	}
}
/*int main() {
}*/

Compilation message (stderr)

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:46:19: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   46 |   while (v.size() != n) {
      |          ~~~~~~~~~^~~~
split.cpp:15:19: warning: unused variable 'sb2' [-Wunused-variable]
   15 |  bool sb1 = true, sb2 = true;
      |                   ^~~
split.cpp:59:1: warning: control reaches end of non-void function [-Wreturn-type]
   59 | }
      | ^

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