Submission #785425

#	Time	Username	Problem	Language	Result	Execution time	Memory
785425		JooDdae	Super Dango Maker (JOI22_dango3)	C++17	22 / 100	884 ms	620 KiB

dango3

#include "dango3.h"

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int cnt;
vector<int> type[400];

int query(vector<int> v) {
    for(int i=0;i<cnt;i++) for(int j=1;j<type[i].size();j++) v.push_back(type[i][j]);
    return Query(v);
}

void Solve(int N, int M) {
    vector<int> v(N*M);
    iota(v.begin(), v.end(), 1);

    random_device rd;
    shuffle(v.begin(), v.end(), mt19937(rd()));

    for(int i=0;i<N*M;i++) {
        int u = v.back(); v.pop_back();

        int re = query(v);

        if(re == M-1) {
            type[cnt++].push_back(u);
            continue;
        }

        int l = 0, r = cnt-1;
        while(l < r) {
            int mid = (l+r)/2;

            for(int i=l;i<=mid;i++) v.push_back(type[i][0]);
            int re = query(v);
            for(int i=l;i<=mid;i++) v.pop_back();

            if(re == M-1) r = mid;
            else l = mid+1;
        }
        
        type[l].push_back(u);
    }


    type[N-1] = v;

    for(int i=0;i<M;i++) {
        vector<int> re;
        for(int j=0;j<N;j++) {
            re.push_back(type[j][i]);
        }
        Answer(re);
    }
}

Compilation message (stderr)

dango3.cpp: In function 'int query(std::vector<int>)':
dango3.cpp:11:41: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   11 |     for(int i=0;i<cnt;i++) for(int j=1;j<type[i].size();j++) v.push_back(type[i][j]);
      |                                        ~^~~~~~~~~~~~~~~

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