Submission #783086

#	Time	Username	Problem	Language	Result	Execution time	Memory
783086		JooDdae	Super Dango Maker (JOI22_dango3)	C++17	7 / 100	169 ms	540 KiB

dango3

#include "dango3.h"

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

vector<int> done;

int query(vector<int> q) {
    for(auto x : done) q.push_back(x);
    return Query(q);
}

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()));

    vector<vector<int>> type(N, vector<int>());

    for(int i=0;i<N-1;i++) {
        // cout << i << " " << v.size() << endl;

        int l = 1, r = (int)v.size()-1;
        while(l <= r) {
            int mid = (l+r) >> 1;

            auto re = query(vector<int>(v.begin(), v.begin()+mid+1));
            // cout << mid << " " << re << endl;
            if(re) r = mid-1;
            else l = mid+1;
        }
        // cout << l << " " << r << "]" << endl;

        vector<int> x = vector<int>(v.begin(), v.begin()+l);

        for(int u=l;u<v.size();u++) {
            x.push_back(v[u]);
            if(query(x)) {
                type[i].push_back(v[u]);
                // cout << v[u] << " ";
                v.erase(v.begin()+u), u--;
            }
            x.pop_back();
        }

        for(auto x : type[i]) done.push_back(x);
        // cout << "DONE------------\n";
    }

    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 'void Solve(int, int)':
dango3.cpp:39:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |         for(int u=l;u<v.size();u++) {
      |                     ~^~~~~~~~~

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