题面
题意
交互题
有n个硬币,每个硬币有一个颜色,这个颜色是R或G或B,每次询问可以询问多对硬币是否相同(一个硬币不能同时属于两对),最多7次询问,将硬币按颜色分成三对.
做法
首先考虑硬币只有两种颜色怎么做,发现可以只用两次询问得到:
第一次询问:1 2|3 4|5 6|…
第二次询问:2 3|4 5|6 7|…
这样就可以得到标号相邻的硬币的颜色是否相同,即可得到每个硬币的颜色.
考虑三种颜色怎么做,先将颜色相同的硬币合并,这样就可以得到一个相邻两硬币颜色不同的序列num.然后再用两次询问求出
n
u
m
[
i
]
num[i]
num[i]与
n
u
m
[
i
+
2
]
num[i+2]
num[i+2]是否相同,即可得到答案.
代码
#include<bits/stdc++.h>
#define P pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define N 100100
using namespace std;
int T,n,top,num[N],ans[N];
char a[N];
vector<int>A,B,C;
map<P,bool>mm;
inline void ask()
{
int i,j;
if(!A.size()) return;
printf("Q %d ",(int)A.size());
for(i=0;i<A.size();i++) printf("%d %d ",A[i],B[i]);puts("");
fflush(stdout);
scanf("%s",a+1);
for(i=1;i<=A.size();i++) mm[mp(A[i-1],B[i-1])]=(a[i]=='1');
}
int main()
{
int i,j,t;
cin>>T;
while(T--)
{
top=0,mm.clear();
memset(ans,0,sizeof(ans));
scanf("%d",&n);
A.clear(),B.clear();
for(i=2;i<=n;i+=2)
{
A.push_back(i-1);
B.push_back(i);
}
ask();
A.clear(),B.clear();
for(i=3;i<=n;i+=2)
{
A.push_back(i-1);
B.push_back(i);
}
ask();
num[++top]=1;
for(i=2;i<=n;i++) if(!mm[mp(i-1,i)]) num[++top]=i;
A.clear(),B.clear();
for(i=3;i<=top;i+=4) A.push_back(num[i-2]),B.push_back(num[i]);
for(i=4;i<=top;i+=4) A.push_back(num[i-2]),B.push_back(num[i]);
ask();
A.clear(),B.clear();
for(i=5;i<=top;i+=4) A.push_back(num[i-2]),B.push_back(num[i]);
for(i=6;i<=top;i+=4) A.push_back(num[i-2]),B.push_back(num[i]);
ask();
ans[num[1]]=1;
if(top>1) ans[num[2]]=2;
for(i=3;i<=top;i++)
{
if(mm[mp(num[i-2],num[i])]) ans[num[i]]=ans[num[i-2]];
else ans[num[i]]=6-ans[num[i-2]]-ans[num[i-1]];
}
A.clear(),B.clear(),C.clear();
for(i=1;i<=n;i++)
{
if(ans[i]) t=ans[i];
ans[i]=t;
if(ans[i]==1) A.push_back(i);
else if(ans[i]==2) B.push_back(i);
else if(ans[i]==3) C.push_back(i);
}
printf("A %d %d %d\n",(int)A.size(),(int)B.size(),(int)C.size());
for(i=0;i<A.size();i++) printf("%d ",A[i]);puts("");
for(i=0;i<B.size();i++) printf("%d ",B[i]);puts("");
for(i=0;i<C.size();i++) printf("%d ",C[i]);puts("");
fflush(stdout);
}
}