hiho一下 第三十周 小Hi小Ho的惊天大作战：扫雷·一

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

故事背景：密室、监视器与充满危机的广场

“我们还是循序渐进，先来考虑这样一个简单化问题：”小Hi思索片刻，道：“在一个大小为2*N的广场，其中第一行里的某一些格子里可能会有至多一个地雷，而第二行的格子里全都为数字，表示第一行中距离与这个格子不超过2的格子里总共有多少个地雷，即第二行的第i个格子里的数字表示第一行的第i-1个, 第i个, 第i+1个，三个格子（如果i=1或者N则不一定有三个）里的地雷的总数。”

“而我们要做的是——找出哪些地方一定是雷，哪些地方一定不是雷。”小Ho道：“不然，我可就要光荣牺牲了。”

提示：寻找关键点——那些一旦决定之后就能让局面豁然开朗的地方。

输入

每个测试点（输入文件）存在多组测试数据。

每个测试点的第一行为一个整数Task，表示测试数据的组数。

在一组测试数据中：

第1行为1个整数N，表示迷宫的宽度。

第2行为N个整数A_1 ... A_N，依次表示迷宫第二行的N个格子里标注的数字。

对于100%的数据，满足1<=N<=10^5, 0<=a_i<=3.<>

对于100%的数据，满足符合数据描述的地图一定存在。

输出

对于每组测试数据，输出2行，其中第一行先输出一定为地雷的格子的数量，然后按照从小到大的顺序输出所有一定为地雷的格子的位置，第二行先输出一定不为地雷的格子的数量，按照从小到大的顺序输出所有一定不为地雷的格子的位置。

样例输入

2
3
1 1 1
10
1 2 1 2 2 3 2 2 2 2 

样例输出

1 2
2 1 3
7 1 3 5 6 7 9 10
3 2 4 8 

思想：用递归枚举所有可能的答案。

#include<cstdio>
#include<stdlib.h>
#include<string.h>
//using namespace std;
#define N 100005
//#pragma warning(disable:4996)
int X[N];	//地雷
int A[N];	//记录地雷数目的数组
int S[N];	//标记格子是否确定一定是地雷或一定不是地雷
int ans;
int YES[N];
int NO[N];

int check(int n,int cur)
{
	if (cur == 2 && X[1] + X[2] == A[1])
	{
		if (n == 2 && X[1] + X[2] == A[2])
			return 1;
		if (n != 2)
			return 1;
	}
	if (cur == n && X[n] + X[n - 1] + X[n - 2] == A[n - 1] && X[n] + X[n - 1] == A[n])
		return 1;
	if (cur != n && X[cur] + X[cur - 1] + X[cur - 2] == A[cur - 1])
		return 1;
	return 0;
}

void mine(int n,int cur)
{
	if (n == 1)
	{
		S[1] = A[1]; return;
	}
	if (cur > n)
	{
		if (!ans)
		{
			ans++;
			memcpy(S + 1, X + 1, n*sizeof(int));
		}
		else
		{
			ans++;
			for (int i = 1; i <= n; i++)
			{
				if (S[i] != X[i])
					S[i] = -1;		//这个位置不确定。。。
			}
		}
		return;
	}
	X[cur] = 0;
	if(cur == 1 || check(n, cur))
		mine(n, cur + 1);
	X[cur] = 1;
	if (cur == 1 || check(n, cur))
		mine(n, cur + 1);
}

void print_ans(int k)
{
	int y=0, n=0;
	for (int i = 1; i <= k; i++)
	{
		if (S[i] == 0)
		{
			n++; NO[n] = i;
		}
		if (S[i] == 1)
		{
			y++; YES[y] = i;
		}
	}
	printf("%d ", y);
	for (int i = 1; i <= y; i++) printf("%d ", YES[i]);
	printf("\n");
	printf("%d ", n);
	for (int i = 1; i <= n; i++) printf("%d ", NO[i]);
	printf("\n");
}

int main()
{
	int t,n;
	scanf("%d", &t);
	for (int i = 0; i < t; i++)
	{
		scanf("%d", &n);
		ans = 0;
		for (int j = 1; j <= n; j++)
		{
			scanf("%d", &A[j]);
		}
		mine(n,1);
		print_ans(n);
	}
	//system("pause");
	return 0;
}