题目1 : 小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
//
// hihoweek30.cpp
// HDOJ
//
// Created by cipher on 15/1/28.
// Copyright (c) 2015年 cipher. All rights reserved.
//
#include<iostream>
#include <stdio.h>
#include<stdlib.h>
#include<string.h>
using namespace std;
#define N 100005
int num[N],x[N],sure[N];//雷的数量，雷，确定是雷
int yes[N],no[N];//输出 有雷的，无雷的
int n,f;
bool check(int n,int cur)
{
//当前第2个格子
if(cur == 2 && x[1]+x[2] == num[1])
{
if(n==2 && x[1]+x[2]==num[2])
return true;
if(n!=2)
return true;
}
//判断最后个格子
if(cur == n && x[n-2]+x[n-1]+x[n]==num[n-1] && x[n-1]+x[n]==num[n])
return true;
//前一个满足
if(cur!=n && x[cur-2]+x[cur-1]+x[cur]==num[cur-1])
return true;
return false;
}
void solve(int n,int cur)
{
if(n==1)
{
sure[1]=num[1];
return ;
}
if(cur > n)
{
if(!f)//只有一次判断
{
f=1;
memcpy(sure+1, x+1, n*sizeof(int));//拷贝雷的数目给确定点
//src copy to des
//void *memcpy(void *dest, const void *src, size_t n);
}
else
{
for(int i=1;i<=n;i++)
{
if(sure[i]!=x[i])//不同的，不确定的
sure[i]=-1;
}
}
return ;
}
x[cur]=0;
if(cur==1 || check(n,cur))
solve(n, cur+1);
x[cur]=1;
if(cur==1 || check(n,cur))
solve(n, cur+1);
}
//打印
void print()
{
int y=0,w=0;
for(int i=1;i<=n;i++)
{
if(sure[i]==0)
{
no[w++]=i;
}
if(sure[i]==1)
{
yes[y++]=i;
}
}
printf("%d",y);
for(int i=0;i<y;i++)
{
printf(" %d",yes[i]);
}
printf("\n");
printf("%d",w);
for(int i=0;i<w;i++)
{
printf(" %d",no[i]);
}
printf("\n");
}
int main()
{
int t,i;
scanf("%d",&t);
while (t--) {
scanf("%d",&n);
for(i=1;i<=n;i++)
{
scanf("%d",&num[i]);
}
f=0;
solve(n,1);
print();
}
return 0;
}
