Submission #580852

#	Time	Username	Problem	Language	Result	Execution time	Memory
580852		JustInCase	Super Dango Maker (JOI22_dango3)	C++17	100 / 100	3708 ms	528 KiB

dango3

#include "dango3.h"
#include <random>
#include <vector>
#include <algorithm>

#define solve Solve
#define	submitAnswer Answer
#define query Query

std::mt19937 mt(69);

void solve(int32_t n, int32_t m) {
	std::vector< int32_t > remDangos(n * m);
	std::vector< bool > used(n * m + 1, false);
	for(int32_t i = 1; i <= n * m; i++) {
		remDangos[i - 1] = i;
	}

	for(int32_t r = 0; r < m; r++) {
		shuffle(std::begin(remDangos), std::end(remDangos), mt);
		
		std::vector< int32_t > currentDangos;
		for(auto &x : remDangos) {
			std::vector< int32_t > queryDangos;
			for(int32_t i = 1; i <= n * m; i++) {
				if(!used[i] && i != x) {
					queryDangos.push_back(i);
				}
			}

			if(query(queryDangos) == m - r - 1) {
				currentDangos.push_back(x);
				used[x] = true;

				if(currentDangos.size() == n) {
					break;
				}
			}
		}

		submitAnswer(currentDangos);
		
		std::vector< int32_t > newRem;
		for(int32_t i = 1; i <= n * m; i++) {
			if(!used[i]) {
				newRem.push_back(i);
			}
		}

		remDangos = newRem;
	}
}

Compilation message (stderr)

dango3.cpp: In function 'void Solve(int32_t, int32_t)':
dango3.cpp:35:29: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int32_t' {aka 'int'} [-Wsign-compare]
   35 |     if(currentDangos.size() == n) {
      |        ~~~~~~~~~~~~~~~~~~~~~^~~~

• Subtask 1: 2 / 2
• Subtask 2: 5 / 5
• Subtask 3: 15 / 15
• Subtask 4: 78 / 78