hihoCoder 1114 扫雷1

题意 在一个大小为2*N的广场，其中第一行里的某一些格子里可能会有至多一个地雷，而第二行的格子里全都为数字，表示第一行中距离与这个格子不超过2的格子里总共有多少个地雷，即第二行的第i个格子里的数字表示第一行的第i-1个, 第i个, 第i+1个，三个格子（如果i=1或者N则不一定有三个）里的地雷的总数。找出哪些地方一定是雷，哪些地方一定不是雷。

思路 这题被卡智商了...还以为是人工智能的推理呢.....额....

   其实确定了第一个格放没放雷,根据已知后面的位置完全都能确定了...只需要枚举第一个格的情况，递推出后面的情况，看后面格子如果出现只有1种情况的，则可以确定是否放雷，否则不能确定。

#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;
}