1114

#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 

以前也做过这种类型的题目,这种类型就是第一个格子定了其他的格子也定了，思路很简单，就是注意细节。

题意：有三种情况

1.第一个格子只能放雷，这种情况下输出一定是雷的个数和位置，一定不是雷的个数和位置；

2.第一个格子不能放雷，这种情况下输出一定是雷的个数和位置，一定不是雷的个数和位置；

3.第一个格子放雷和不放雷都满足，这种情况下，就是在一定是雷的里面找共同的，在一定不是雷的位置里找共同的。

细节就是

1.一定是雷或一定不是雷的个数为0时，位置就不输出，没有怎么输出

2.n==1的时候特判，我的代码是这样的因为我是i=1;i<n;

#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn = 100005;
int a[maxn];
int f[maxn][2];
bool flag[2];
int n;
queue<int> q[2];

void print()
{
	for(int i=1;i>=0;i--)
	{
		cout<<q[i].size();
		while(q[i].size() != 0)
		{
			int tmp = q[i].front();
			q[i].pop();
			cout<<' '<<tmp;
		}
		puts("");
	}

}

int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
		}
		flag[0] = flag[1] = false;
		f[1][0] = 0;
		f[1][1] = 1;

		for(int j=0;j<2;j++)
		{
			int sum = (int)f[1][j];
			for(int i=2;i<=n;i++)
			{
				f[i][j] = a[i-1] - sum;
				if(f[i][j] >= 2 || f[i][j] < 0)
				{
					flag[j] = true;
					break;
				}
				sum -= f[i-2][j] - f[i][j];
			}
			if(f[n-1][j] + f[n][j] != a[n])
			{
				flag[j] = true;
			}
		}
		for(int j=0;j<2;j++)
		{
			if(flag[j] == true)
			{
				for(int i=1;i<=n;i++)
					q[f[i][j^1]].push(i);
			}
		}
		if(flag[0]==1 || flag[1]==1)
		{
			print();
			continue;
		}
		for(int i=1;i<=n;i++)
		{
			if(f[i][0] != f[i][1])
				continue;
			q[f[i][0]].push(i);
		}
		print();
	}
	return 0;
}