#include<bits/stdc++.h>
using namespace std;
bool ask(vector<int> v)
{
cout << "query ";
for(int i : v)
cout << i << ' ';
cout << endl;
bool res;
cin >> res;
return res;
}
void out(vector<int> a, vector<int> b)
{
cout << "end\n";
for(int i : a)
cout << i << ' ';
cout << endl;
for(int i : b)
cout << i << ' ';
cout << endl;
exit(0);
}
vector<int> compute(vector<vector<int> > G, int m)
{
int n = G.size();
int in[n] = {};
for(int i = 0; i < n; i ++)
for(int j : G[i])
{
// cerr << i << ' ' << j << endl;
in[j]++;
}
vector<int> topo;
set<int> st;
for(int i = 0; i < n; i ++)
if(in[i] == 0)
st.insert(i * m);
while(st.size())
{
int u = *st.begin();
//cerr << u << endl;
u /= m;
st.erase(st.begin());
topo.push_back(u);
for(int v : G[u])
{
in[v]--;
if(in[v] == 0)
st.insert(v * m);
}
}
vector<int> ans(n);
for(int i = 0; i < n; i++)
ans[topo[i]] = i + 1;
return ans;
}
int main()
{
int n;
cin >> n;
vector<int> p(n), idx(n + 1);
for(int i = 0; i < n; i ++)
{
cin >> p[i];
idx[p[i]] = i;
}
vector<vector<int> > G(n), I(n);
vector<pair<int,int> > conditions;
bool f[n] = {};
for(int v = 1; v <= n; v++)
{
vector<int> vec = {idx[v]};
for(int u = v - 1; u > 0; u--)
{
vector<int> q = p;
vec.push_back(idx[u]);
for(int j = 0; j + 1 < vec.size(); j++)
swap(q[vec[j]], q[vec[j + 1]]);
int cur = vec.back();
while(G[cur].size())
{
// cerr << "cur = " << cur << endl;
int i = G[cur][0];
for(int k : G[cur])
if(q[i] < q[k])
i = k;
swap(q[i], q[cur]);
cur = i;
}
bool res = ask(q);
if(!res)
{
conditions.push_back({vec.back(), vec[0]});
G[vec.back()].push_back(vec[0]);
I[vec[0]].push_back(vec.back());
// cerr << vec[0] << ' ' << vec.back() << endl;
}
}
}
/*
for(auto [x, y] : conditions)
{
// G[x].push_back(y), I[y].push_back(x);
cerr << x << ' ' << y << endl;
}
// */
vector<int> a = compute(G, 1);
vector<int> b = compute(G, -1);
out(a, b);
return 0;
}
Compilation message
zagonetka.cpp: In function 'int main()':
zagonetka.cpp:91:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
91 | for(int j = 0; j + 1 < vec.size(); j++)
| ~~~~~~^~~~~~~~~~~~
zagonetka.cpp:83:8: warning: unused variable 'f' [-Wunused-variable]
83 | bool f[n] = {};
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
344 KB |
Output is correct |
3 |
Correct |
0 ms |
344 KB |
Output is correct |
4 |
Correct |
0 ms |
344 KB |
Output is correct |
5 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
356 KB |
Output is correct |
2 |
Incorrect |
12 ms |
428 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
344 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
37 ms |
412 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |