#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 

解题思路：利用dfs进行暴力寻找。

具体代码如下：

#include<iostream>
#include<stdio.h>
using namespace std;
int k;
int s;//有几种情况
int logal[105][100010];
int ans1[100010],ans2[100010];
int d[100010];
int num[100010];
void dfs(int n)
{
    if(n==k+1)
    {
        if(num[k]==d[k]+d[k-1])
        {s++;
        for(int i=1;i<=k;i++)
        {
            logal[s][i]=d[i];//第s种情况的第i个位置有无地雷
        }
        return;
        }
    }
    if(num[n-1]==d[n-1]+d[n-2])
    {
        d[n]=0,dfs(n+1);
    }
    else if(num[n-1]==d[n-1]+d[n-2]+1)
    {
        d[n]=1,dfs(n+1);
    }
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        s=0;
        scanf("%d",&k);
        for(int i=1;i<=k;i++)
        {
            cin>>num[i];
        }
        d[0]=0,d[1]=0;
        dfs(2);
        d[0]=0,d[1]=1;
        dfs(2);
        int num_0=0,num_1=0;
        if(s==1)
        {
            for(int i=1;i<=k;i++)
            {
                if(logal[1][i]==1)
                    ans1[num_1++]=i;
                else if(logal[1][i]==0)
                    ans2[num_0++]=i;
            }
        }
        else
        {
            for(int i=1;i<=k;i++)
            {
                if(logal[1][i]==1&&logal[2][i]==1)
                    ans1[num_1++]=i;
                else if(logal[1][i]==0&&logal[2][i]==0)
                    ans2[num_0++]=i;
            }
        }
        printf("%d ",num_1);
        for(int i=0;i<num_1;i++)
        {
            printf("%d ",ans1[i]);
        }
        cout<<endl;
        printf("%d ",num_0);
        for(int i=0;i<num_0;i++)
        {
            printf("%d ",ans2[i]);
        }
        cout<<endl;

    }
    return 0;
}