Submission #951743

#	Time	Username	Problem	Language	Result	Execution time	Memory
951743		pcc	Super Dango Maker (JOI22_dango3)	C++17	100 / 100	131 ms	616 KiB

dango3

#include "dango3.h"

#include <bits/stdc++.h>
#include <vector>
using namespace std;

namespace {

const int mxn = 1e5+10;
vector<int> perm;
bitset<mxn> done;

int ask(vector<int> vv){
	//cerr<<"ASK:";for(auto &i:vv)cerr<<i<<' ';cerr<<endl;
	return Query(vv);
}

}  // namespace
/*
std::vector<int> x(3);
x[0] = 1;
x[1] = 2;
x[2] = 3;
variable_example = Query(x);
for (int i = 0; i < M; i++) {
std::vector<int> a(N);
for (int j = 0; j < N; j++) {
  a[j] = N * i + j + 1;
}
Answer(a);
}
*/


void Solve(int N, int M) {
	for(int i = 1;i<=N*M;i++)perm.push_back(i);
	mt19937 seed(time(NULL));
	shuffle(perm.begin(),perm.end(),seed);
	for(int i = 1;i<=M;i++){
		shuffle(perm.begin(),perm.end(),seed);
		int l = 0,r = perm.size()-1;
		vector<int> v;
		while(l != r){
			int mid = (l+r)>>1;
			vector<int> tmp;
			for(int i = 0;i<=mid;i++)tmp.push_back(perm[i]);
			if(tmp.size()<N)l = mid+1;
			else if(ask(tmp) != 0)r = mid;
			else l = mid+1;
		}
		for(int i = 0;i<=l;i++)v.push_back(perm[i]);
		while(v.size()>N){
			auto tmp = v.back();
			v.pop_back();
			if(ask(v) == 0)v.insert(v.begin(),tmp);
			//for(auto &j:v)cerr<<j<<' ';cerr<<endl;
		}
		for(auto &j:v)perm.erase(find(perm.begin(),perm.end(),j));
		//cerr<<"ANS:";for(auto &j:v)cerr<<j<<' ';cerr<<endl;
		Answer(v);
	}
	return;
}

Compilation message (stderr)

dango3.cpp: In function 'void Solve(int, int)':
dango3.cpp:47:17: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   47 |    if(tmp.size()<N)l = mid+1;
      |       ~~~~~~~~~~^~
dango3.cpp:52:17: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   52 |   while(v.size()>N){
      |         ~~~~~~~~^~

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