Submission #423972

#	Time	Username	Problem	Language	Result	Execution time	Memory
423972		schse	Split the Attractions (IOI19_split)	C++17	18 / 100	112 ms	15268 KiB

split

#include "split.h"
#include <bits/stdc++.h>
#ifndef EVAL
#include "grader.cpp"
#endif
using namespace std;

struct node
{
	vector<int> edges;
	bool been;
};

vector<int> res;
vector<int> order;
vector<node> g;

int A, B, C;

void dfs(int n)
{
	if (g[n].been)
		return;
	order.push_back(n);
	g[n].been = true;
	for (int i : g[n].edges)
		dfs(i);
	return;
}

vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q)
{
	A = a, B = b, C = c;
	g.resize(n);
	res.resize(n);

	for (int i = 0; i < p.size(); i++)
	{
		g[p[i]].edges.push_back(q[i]);
		g[q[i]].edges.push_back(p[i]);
	}
	int stp = 0;
	for (stp = 0; stp < n; stp++)
	{
		if (g[stp].edges.size() == 1)
		{
			dfs(stp);
			break;
		}
	}
	if (stp == n)
		dfs(0);

	stp = 0;
	while (a--)
	{
		res[order[stp]] = 1;
		stp++;
	}
	while (b--)
	{
		res[order[stp]] = 2;
		stp++;
	}
	while (c--)
	{
		res[order[stp]] = 3;
		stp++;
	}
	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:37:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |  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