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

#1114 : 小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 

我感觉这个题数据好像有点大，dfs可能会爆内纯，可是看了网上的代码，想想只有dfs了，枚举当前点的状态，然后dfs

具体解释在代码中

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<vector>
#include<map>

#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define MID(x,y) ((x+y)>>1)

#define eps 1e-8
using namespace std;
#define N 100005

int a[N],vis[N],ans[N],n,yes;  //vis是标记有无雷
int YES[N],NO[N];

bool judge(int pos)
{
	if(pos==1)
	{
		if(n==1)
			return vis[pos]==a[pos];
		return a[pos]>=vis[pos];
	}

	if(pos==2)
	{
		int temp=vis[1]+vis[2];
		if(n==2)
			return a[1]==temp&&a[2]==temp;
		return a[1]==temp;
	}

	int temp=vis[pos]+vis[pos-1]+vis[pos-2];

	if(pos==n)
	  return a[pos-1]==temp&&a[pos]==vis[pos]+vis[pos-1];

	return a[pos-1]==temp;
}

void dfs(int pos)
{
	int i,j;

    if(pos==n+1)
	{
		if(!yes)
		{
			yes=1;
			for(i=1;i<=n;i++)   //第一次成功时复制到ans
				ans[i]=vis[i];

			return ;
		}

		for(i=1;i<=n;i++)
			if(ans[i]!=vis[i])   //以后成功的情况看看有雷的地方是不是一定有雷
			  ans[i]=-1;         //如果这种情况有雷,那种情况无雷这个点无法确定
		return ;
	}

	vis[pos]=1;
	if(judge(pos))
		dfs(pos+1);

	vis[pos]=0;
	if(judge(pos))
		dfs(pos+1);
}

void print()
{
	int i,j,x,y;
	x=0,y=0;

	for(i=1;i<=n;i++)
	{
		if(ans[i]==1) YES[x++]=i;
		if(ans[i]==0) NO[y++]=i;
	}

	printf("%d",x);
	for(i=0;i<x;i++)
		printf(" %d",YES[i]);
		printf("\n");

	printf("%d",y);
	for(i=0;i<y;i++)
		printf(" %d",NO[i]);
	printf("\n");
}


int main()
{
	int i,j,t;
	scanf("%d",&t);
	while(t--)
	{
		yes=0;
		scanf("%d",&n);
		for(i=1;i<=n;i++)
			scanf("%d",&a[i]);

		dfs(1);
		print();
	}
	return 0;


}