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行，其中第一行先输出一定为地雷的格子的数量，然后按照从小到大的顺序输出所有一定为地雷的格子的位置，第二行先输出一定不为地雷的格子的数量，按照从小到大的顺序输出所有一定不为地雷的格子的位置。

样例输入

231 1 1101 2 1 2 2 3 2 2 2 2 

样例输出

1 22 1 37 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;}