Submission #1051844

#	Time	Username	Problem	Language	Result	Execution time	Memory
1051844		Tob	Split the Attractions (IOI19_split)	C++14	18 / 100	42 ms	14288 KiB

split

#include <bits/stdc++.h>

#include "split.h"

#define F first
#define S second

using namespace std;

typedef long long ll;

const int N = 1e5 + 7;

int n;
int siz[N];
bool bio[N];
vector <int> no;
vector <int> adj[N];

void ldfs(int x) {
	bio[x] = 1;
	no.push_back(x);
	for (auto y : adj[x]) {
		if (bio[y]) continue;
		ldfs(y);
	}
}

vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
	n = a+b+c;
	vector <int> res(n, 0);
	for (int i = 0; i < p.size(); i++) {
		adj[p[i]].push_back(q[i]);
		adj[q[i]].push_back(p[i]);
	}
	int mxdeg = 0;
	for (int i = 0; i < n; i++) mxdeg = max(mxdeg, (int)adj[i].size());
	if (mxdeg <= 2) {
		ldfs(0);
		for (int i = 0; i < a; i++) res[no[i]] = 1;
		for (int i = a; i < a+b; i++) res[no[i]] = 2;
		for (int i = a+b; i < n; i++) res[no[i]] = 3;
	}
	else if (a == 1) {
		ldfs(0);
		for (int i = 0; i < b; i++) res[no[i]] = 2;
		for (int i = b; i < n-1; i++) res[no[i]] = 3;
		res[no[n-1]] = 1;
	}
	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:32:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |  for (int i = 0; i < p.size(); i++) {
      |                  ~~^~~~~~~~~~

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