#include "dango3.h"
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int MAXN=1e4+10;
bool used[MAXN];
vector <int> solve(vector <int> a, int n, int m) {
if (a.size()==n) {
Answer(a);
return {};
}
if (m>1) {
vector <int> q, s1;
int l, r, mid, gr=m/2;
l=gr*n-1; r=a.size()-1;
while (l<r) {
mid=(l+r)/2;
if (!q.empty()) q.clear();
for (int i=0;i<=mid;i++) q.push_back(a[i]);
if (Query(q)<gr) l=mid+1;
else r=mid;
}
for (int i=0;i<=l;i++) s1.push_back(a[i]);
if (!q.empty()) q.clear();
q=solve(s1,n,gr);
s1.clear();
for (int i=l+1;i<a.size();i++) s1.push_back(a[i]);
for (auto i:q) s1.push_back(i);
return solve(s1,n,m-gr);
}
vector <int> cur_ans, q;
int ind=a.size()-1, gr=0;
cur_ans.push_back(a[ind]);
int l, r, mid;
for (int j=1;j<n;j++) {
l=gr; r=ind-1;
while (l+1<r) {
mid=(l+r)/2;
if (!q.empty()) q.clear();
for (int i=mid;i<ind;i++) q.push_back(a[i]);
for (int i=0;i<j;i++) q.push_back(cur_ans[i]);
if (Query(q)==0) r=mid;
else l=mid;
}
cur_ans.push_back(a[l]);
gr=l+1;
}
for (int i=0;i<n;i++) used[cur_ans[i]]=true;
vector <int> newa;
for (int i=0;i<a.size();i++) {
if (!used[a[i]]) newa.push_back(a[i]);
}
Answer(cur_ans);
return newa;
}
void Solve(int N, int M) {
int n, m;
n=N; m=M;
vector <int> a;
for (int i=1;i<=n*m;i++) a.push_back(i);
//random_shuffle(a.begin(),a.end());
solve(a,n,m);
}
Compilation message
dango3.cpp: In function 'std::vector<int> solve(std::vector<int>, int, int)':
dango3.cpp:9:17: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
9 | if (a.size()==n) {
| ~~~~~~~~^~~
dango3.cpp:30:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
30 | for (int i=l+1;i<a.size();i++) s1.push_back(a[i]);
| ~^~~~~~~~~
dango3.cpp:53:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
53 | for (int i=0;i<a.size();i++) {
| ~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
1 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
348 KB |
Output is correct |
2 |
Correct |
3 ms |
348 KB |
Output is correct |
3 |
Correct |
7 ms |
348 KB |
Output is correct |
4 |
Correct |
9 ms |
348 KB |
Output is correct |
5 |
Correct |
1 ms |
420 KB |
Output is correct |
6 |
Correct |
2 ms |
348 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
33 ms |
604 KB |
Output is correct |
2 |
Incorrect |
24 ms |
600 KB |
Wrong Answer [7] |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
54 ms |
860 KB |
Wrong Answer [7] |
2 |
Halted |
0 ms |
0 KB |
- |