#include "scales.h"
#include <bits/stdc++.h>
#define BS bitset <720>
using namespace std;
struct question{
int a[4];
int type;
};
unordered_map <BS, int> mp[7];
BS p[120][3];
vector <vector <int>> v;
vector <question> vq;
int pw[7];
int ask(int i){
int res=0;
if (!vq[i].type)
res=getLightest(vq[i].a[0],vq[i].a[1],vq[i].a[2]);
if (vq[i].type==1)
res=getMedian(vq[i].a[0],vq[i].a[1],vq[i].a[2]);
if (vq[i].type==2)
res=getHeaviest(vq[i].a[0],vq[i].a[1],vq[i].a[2]);
if (vq[i].type==3)
res=getNextLightest(vq[i].a[0],vq[i].a[1],vq[i].a[2],vq[i].a[3]);
return (res==vq[i].a[0]?0:(res==vq[i].a[1]?1:2));
}
int f(BS b, int id){
if (b.count()>pw[id])
return -1;
if (!id)
return -2;
if (b.count()<2)
return -2;
if (mp[id].count(b))
return mp[id][b];
for (int i=0;i<vq.size();i++){
bool ch=true;
for (int j=0;j<3;j++)
if (f(b&p[i][j],id-1)==-1){
ch=false;
break;
}
if (ch)
return mp[id][b]=i;
}
return mp[id][b]=-1;
}
bool isLightest(vector <int> a, int i, int j, int k){
return (a[i]<min(a[j],a[k]));
}
bool isMedian(vector <int> a, int i, int j, int k){
return (a[j]-a[i])*(a[j]-a[k])<0;
}
bool isHeaviest(vector <int> a, int i, int j, int k){
return (a[i]>max(a[j],a[k]));
}
bool isNextLightest(vector <int> a, int i, int j, int k, int l){
if (max(a[i],max(a[j],a[k]))<a[l])
return isLightest(a,i,j,k);
if (a[i]<a[l])
return false;
return (!((a[j]>a[l]&&a[j]<a[i])||(a[k]>a[l]&&a[k]<a[i])));
}
void init(int T){
v.clear();
vq.clear();
pw[0]=1;
for (int i=1;i<=6;i++)
pw[i]=pw[i-1]*3;
vector <int> ve;
for (int i=1;i<=6;i++)
ve.push_back(i);
do{
v.push_back(ve);
}while (next_permutation(ve.begin(),ve.end()));
int id=0;
for (int i=1;i<=6;i++)
for (int j=i+1;j<=6;j++)
for (int k=j+1;k<=6;k++){
for (int l=0;l<3;l++)
vq.push_back({i,j,k,0,l});
for (int l=1;l<=6;l++)
if (l!=i&&l!=j&&l!=k)
vq.push_back({i,j,k,l,3});
}
for (int i=0;i<vq.size();i++)
for (int j=0;j<v.size();j++){
for (int k=0;k<3;k++){
if (!vq[i].type&&isLightest(v[j],vq[i].a[k]-1,vq[i].a[(k+1)%3]-1,vq[i].a[(k+2)%3]-1))
p[i][k][j]=1;
if (vq[i].type==1&&isMedian(v[j],vq[i].a[(k+1)%3]-1,vq[i].a[k]-1,vq[i].a[(k+2)%3]-1))
p[i][k][j]=1;
if (vq[i].type==2&&isHeaviest(v[j],vq[i].a[k]-1,vq[i].a[(k+1)%3]-1,vq[i].a[(k+2)%3]-1))
p[i][k][j]=1;
if (vq[i].type==3&&isNextLightest(v[j],vq[i].a[k]-1,vq[i].a[(k+1)%3]-1,vq[i].a[(k+2)%3]-1,vq[i].a[3]-1))
p[i][k][j]=1;
}
}
}
void orderCoins(){
int W[] = {1, 2, 3, 4, 5, 6};
BS b;
for (int i=0;i<720;i++)
b[i]=1;
int tmp=f(b,6);
for (int i=6;i>=1;i--){
int x=mp[i][b];
int y=ask(x);
b&=p[x][y];
if (b.count()==1)
break;
}
int pos=0;
for (int i=0;i<720;i++)
if (b[i]){
pos=i;
break;
}
for (int i=0;i<6;i++)
W[v[pos][i]-1]=i+1;
answer(W);
}
Compilation message
scales.cpp: In function 'int f(std::bitset<720>, int)':
scales.cpp:27:18: warning: comparison of integer expressions of different signedness: 'std::size_t' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
27 | if (b.count()>pw[id])
| ~~~~~~~~~^~~~~~~
scales.cpp:35:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<question>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
35 | for (int i=0;i<vq.size();i++){
| ~^~~~~~~~~~
scales.cpp: In function 'void init(int)':
scales.cpp:85:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<question>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
85 | for (int i=0;i<vq.size();i++)
| ~^~~~~~~~~~
scales.cpp:86:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
86 | for (int j=0;j<v.size();j++){
| ~^~~~~~~~~
scales.cpp:75:9: warning: unused variable 'id' [-Wunused-variable]
75 | int id=0;
| ^~
scales.cpp:63:15: warning: unused parameter 'T' [-Wunused-parameter]
63 | void init(int T){
| ~~~~^
scales.cpp: In function 'void orderCoins()':
scales.cpp:104:9: warning: unused variable 'tmp' [-Wunused-variable]
104 | int tmp=f(b,6);
| ^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
231 ms |
9128 KB |
Output is correct |
2 |
Correct |
257 ms |
9288 KB |
Output is correct |
3 |
Correct |
244 ms |
9224 KB |
Output is correct |
4 |
Correct |
240 ms |
9128 KB |
Output is correct |
5 |
Correct |
230 ms |
9220 KB |
Output is correct |
6 |
Correct |
227 ms |
9180 KB |
Output is correct |
7 |
Correct |
229 ms |
9204 KB |
Output is correct |
8 |
Correct |
229 ms |
9156 KB |
Output is correct |
9 |
Correct |
255 ms |
9300 KB |
Output is correct |
10 |
Correct |
246 ms |
9148 KB |
Output is correct |
11 |
Correct |
235 ms |
9244 KB |
Output is correct |
12 |
Correct |
247 ms |
9196 KB |
Output is correct |
13 |
Correct |
236 ms |
9192 KB |
Output is correct |
14 |
Correct |
237 ms |
9160 KB |
Output is correct |
15 |
Correct |
237 ms |
9180 KB |
Output is correct |
16 |
Correct |
234 ms |
9188 KB |
Output is correct |
17 |
Correct |
251 ms |
9192 KB |
Output is correct |
18 |
Correct |
226 ms |
9216 KB |
Output is correct |
19 |
Correct |
265 ms |
9224 KB |
Output is correct |
20 |
Correct |
236 ms |
9216 KB |
Output is correct |
21 |
Correct |
238 ms |
9220 KB |
Output is correct |
22 |
Correct |
240 ms |
9192 KB |
Output is correct |
23 |
Correct |
234 ms |
9160 KB |
Output is correct |
24 |
Correct |
244 ms |
9248 KB |
Output is correct |
25 |
Correct |
247 ms |
9140 KB |
Output is correct |
26 |
Correct |
251 ms |
9248 KB |
Output is correct |
27 |
Correct |
239 ms |
9252 KB |
Output is correct |
28 |
Correct |
233 ms |
9152 KB |
Output is correct |
29 |
Correct |
234 ms |
9204 KB |
Output is correct |
30 |
Correct |
224 ms |
9244 KB |
Output is correct |
31 |
Correct |
232 ms |
9184 KB |
Output is correct |
32 |
Correct |
242 ms |
9372 KB |
Output is correct |
33 |
Correct |
230 ms |
9220 KB |
Output is correct |
34 |
Correct |
232 ms |
9208 KB |
Output is correct |
35 |
Correct |
231 ms |
9220 KB |
Output is correct |
36 |
Correct |
259 ms |
9140 KB |
Output is correct |
37 |
Correct |
244 ms |
9164 KB |
Output is correct |
38 |
Correct |
271 ms |
9220 KB |
Output is correct |
39 |
Correct |
268 ms |
9172 KB |
Output is correct |
40 |
Correct |
245 ms |
9212 KB |
Output is correct |