CF2183G - Snake Instructions

题目链接

题目给的 3 次询问可以分别问 L、R、LR。从 L 到 LR 并不会出现新的伤亡，可以对比位置差异得出没死的蛇的 \(s _ i\)。

现在没死的蛇都可以被分成 0、1、2 三种，死了的蛇设为 ?，则每个 ? 为 1 或 2。若出现 0?0 必定无解，因为无法区分 ? 是 1 还是 2。否则一个 ? 是 1 只有可能是 0? 的情况，这种情况可以通过 R 操作判断。

#include<cstdio>
#include<vector>
using namespace std;

int T,n;
vector<int> ask(const char *s) {
	printf("? %s\n",s),fflush(stdout);
	int k=0; scanf("%d",&k);
	vector<int> res(k);
	for(auto &u:res) scanf("%d",&u);
	return res;
}
void fine(vector<int> s) {
	printf("! ");
	for(auto u:s) printf("%d ",u);
	puts(""),fflush(stdout);
}
void fail() {
	puts("! -1"),fflush(stdout);
}
int main() {
	scanf("%d",&T);
	XJX:while(T--) {
		scanf("%d",&n);
		vector<int> a(n),s(n,-1),p(n*4+2,-1);
		for(int i=0;i<n;i++) scanf("%d",&a[i]),p[a[i]]=i;
		vector<int> L=ask("L"),R=ask("R"),LR=ask("LR");
		int k=LR.size();
		for(int i=0;i<k;i++) s[p[LR[i]]]=LR[i]-L[i];
		for(int i=1;i<n-1;i++) {
			if(a[i-1]==a[i]-1&&a[i+1]==a[i]+1&&s[i-1]==0&&s[i]==-1&&s[i+1]==0) {
				fail(); goto XJX;
			}
		}
		vector<bool> occ(n*4+5);
		for(auto u:R) occ[u]=1;
		for(int i=0;i<n;i++) if(s[i]==-1) {
			if(~p[a[i]-1]&&s[p[a[i]-1]]==0) s[i]=occ[a[i]+1]?1:2;
			else s[i]=2;
		}
		fine(s);
	}
	return 0;
}