hihocoder #1114 扫雷一

时间限制: 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 

解题：

        根据题目提示，使用枚举方法，枚举第一个格子是否有地雷，推算出后面所有格子的情况。每个格子的内容应该是0<=b[i]<=1,需要注意，有可能两种枚举可能都行的同，此时要针对两次计算结果，确认哪些一定是地雷，哪些一定不是地雷。

       我在计算b[i]的时候是分情况讨论的，之前忘了讨论长度为1的情况，提交一直出错，在此提醒自己，谨记谨记！！！！！！！！！

      提示，因为输出规则的限制，输出时做了很多无用功，看到别人用栈的方式来处理输出，很便捷。

#include<iostream>
#include<stdio.h>
using namespace std;
int main()
{
	int num,length,i,sum,j,success,flag;
	int a[100005],b[2][100005];
	freopen("input.txt","r",stdin);
	cin>>num;
	while(num--)
	{
		cin>>length;
		for(i=0;i<length;i++)
			cin>>a[i];
        flag=3;
        success=0;
		b[0][0]=1;
		b[1][0]=0;
		for(j=0;j<2;j++)
		{
			for(i=0;i<length;i++)
			{
			    if(length==1)
                {
                     if(b[j][i]!=a[i])break;
                }
                else
                {
				if(i==0)
				{
					b[j][i+1]=a[i]-b[j][i];
					if(b[j][i+1]>1||b[j][i+1]<0)break;
				}
				else if(i==length-1)
				{
					if(b[j][i]!=(a[i]-b[j][i-1]))break;
				}
				else
				{
					b[j][i+1]=a[i]-b[j][i]-b[j][i-1];
					if(b[j][i+1]>1||b[j][i+1]<0)break;
				}
                }
			}
			if(i==length)
			{
				success++;
				flag=j;
			}
		}
		if(success==2)
		{
			sum=0;
			for(i=0;i<length;i++)
			{
				if((b[0][i]==1)&&(b[1][i]==1))sum++;
			}
			cout<<sum;
			for(i=0;i<length;i++)
			{
				if((b[0][i]==1)&&(b[1][i]==1))
				cout<<' '<<i+1;
			}
			cout<<endl;
			sum=0;
			for(i=0;i<length;i++)
			{
				if((b[0][i]==0)&&(b[1][i]==0))sum++;
			}
			cout<<sum;
			for(i=0;i<length;i++)
			{
				if((b[0][i]==0)&&(b[1][i]==0))
				cout<<' '<<i+1;
			}
			cout<<endl;
		}
		else
		{
			sum=0;
			for(i=0;i<length;i++)
			{
				if(b[flag][i]==1)sum++;
			}
			cout<<sum;
			for(i=0;i<length;i++)
			{
				if(b[flag][i]==1)
					cout<<' '<<i+1;
			}
			cout<<endl;
			cout<<length-sum;
			for(i=0;i<length;i++)
			{
				if(b[flag][i]==0)
					cout<<' '<<i+1;
			}
			cout<<endl;
		}

	}
	return 0;
}