Submission #816994

#	Time	Username	Problem	Language	Result	Execution time	Memory
816994		vjudge1	Split the Attractions (IOI19_split)	C++17	7 / 100	58 ms	10316 KiB

split

#include <bits/stdc++.h>
#include "split.h"

using namespace std;

vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
	vector<int> res(n);
	vector<int> g[n + 2];
	for (int i = 0; i < p.size(); i++) {
		g[p[i]].push_back(q[i]);
		g[q[i]].push_back(p[i]);
	}
	int st = 0;
	for (int i = 0; i < n; i++) {
		if (g[i].size() == 1) {
			st = i;
			break;
		} 
	}
	queue<int> qq;
	qq.push(st);
	int ptr = 0, Left[4] = {a, b, c, 100000};
	bool visited[n + 2];
	memset(visited, 0, sizeof(visited));
	visited[st] = 1;
	res[st] = 1;
	Left[0]--;
	while (qq.size()) {
		if (Left[ptr] == 0) ptr++;
		int fr = qq.front();
		qq.pop();
		for (auto x : g[fr]) {
			if (!visited[x]) {
				visited[x] = 1;
				res[x] = ptr + 1;
				Left[ptr]--;
				if (Left[ptr] == 0) ptr++;
				qq.push(x);
			}
		}
	}
	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:9:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |  for (int i = 0; i < p.size(); i++) {
      |                  ~~^~~~~~~~~~

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