hiho一下第三十周 #1114 小Hi小Ho的惊天大作战：扫雷·一【枚举】

#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 

思路：

枚举第一个-.-

0.0

但是数据应该有BUG。。。

当n=1，a[1]>1    

n=2,a[1]>2

时会出现BUG。。。。

下面代码1和图1为AC的。。。代码2和图2为WA的=====

欢迎测试。。。

其他情况代码1和代码2是一样的呀-.-

但是上面的情况不满足：对于100%的数据，满足符合数据描述的地图一定存在。

我要痴呆了。。。

代码1：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int t,a[100100],shu[10][100100];
bool fafe[10],ans1[100100],ans2[100100];
int main()
{
    while (~scanf("%d",&t))
    {
      //  printf("ru   \\\\\n");
        while (t--)
        {
            int n;scanf("%d",&n);
            for (int i=1;i<=n;i++)
            {
                scanf("%d",&a[i]);
            }
          /*  if (n==1)
            {
                if (a[1])
                printf("1 1 \n0 \n");
                else
                printf("0 \n1 1 \n");
                continue;
            }
            if (n==2)
            {
                if (a[1])
                    printf("0 \n0 \n");
                else
                    printf("0 \n2 1 2 \n");
                continue;
            }*/
            int p=a[1];
            memset(shu,0,sizeof(shu));
            memset(fafe,false,sizeof(fafe));
            memset(ans1,false,sizeof(ans1));
            memset(ans2,false,sizeof(ans2));
            for (int i=0;i<=p;i++)
            {
                shu[i][1]=i;
                shu[i][2]=p-i;
                for (int j=2;j<n;j++)
                    shu[i][j+1]=a[j]-shu[i][j-1]-shu[i][j];
                if (shu[i][n]+shu[i][n-1]==a[n])
                {
                    fafe[i]=true;
                    for (int j=1;j<=n;j++)
                    if (shu[i][j]!=1&&shu[i][j]!=0)
                    {
                        fafe[i]=false;
                        break;
                    }
                }
            }
            for (int i=0;i<=p;i++)
            {
                if (fafe[i])
                {
                    for (int j=1;j<=n;j++)
                    {
                        if (shu[i][j])
                            ans1[j]=true;
                        else
                            ans2[j]=true;
                    }
                }
            }
            int c=0,b=0;
            for (int i=1;i<=n;i++)
            {
                if (!ans2[i])
                    c++;
                if (!ans1[i])
                    b++;
            }
            printf("%d ",c);
            for (int i=1;i<=n;i++)
                if (!ans2[i])
                    printf("%d ",i);
                printf("\n");
            printf("%d ",b);
            for (int i=1;i<=n;i++)
                if (!ans1[i])
                    printf("%d ",i);
                printf("\n");
        }
    }
    return 0;
}

图1.

代码2：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int t,a[100100],shu[10][100100];
bool fafe[10],ans1[100100],ans2[100100];
int main()
{
    while (~scanf("%d",&t))
    {
      //  printf("ru   \\\\\n");
        while (t--)
        {
            int n;scanf("%d",&n);
            for (int i=1;i<=n;i++)
            {
                scanf("%d",&a[i]);
            }
            if (n==1)
            {
                if (a[1])
                printf("1 1 \n0 \n");
                else
                printf("0 \n1 1 \n");
                continue;
            }
            if (n==2)
            {
                if (a[1])
                    printf("0 \n0 \n");
                else
                    printf("0 \n2 1 2 \n");
                continue;
            }
            int p=a[1];
            memset(shu,0,sizeof(shu));
            memset(fafe,false,sizeof(fafe));
            memset(ans1,false,sizeof(ans1));
            memset(ans2,false,sizeof(ans2));
            for (int i=0;i<=p;i++)
            {
                shu[i][1]=i;
                shu[i][2]=p-i;
                for (int j=2;j<n;j++)
                    shu[i][j+1]=a[j]-shu[i][j-1]-shu[i][j];
                if (shu[i][n]+shu[i][n-1]==a[n])
                {
                    fafe[i]=true;
                    for (int j=1;j<=n;j++)
                    if (shu[i][j]!=1&&shu[i][j]!=0)
                    {
                        fafe[i]=false;
                        break;
                    }
                }
            }
            for (int i=0;i<=p;i++)
            {
                if (fafe[i])
                {
                    for (int j=1;j<=n;j++)
                    {
                        if (shu[i][j])
                            ans1[j]=true;
                        else
                            ans2[j]=true;
                    }
                }
            }
            int c=0,b=0;
            for (int i=1;i<=n;i++)
            {
                if (!ans2[i])
                    c++;
                if (!ans1[i])
                    b++;
            }
            printf("%d ",c);
            for (int i=1;i<=n;i++)
                if (!ans2[i])
                    printf("%d ",i);
                printf("\n");
            printf("%d ",b);
            for (int i=1;i<=n;i++)
                if (!ans1[i])
                    printf("%d ",i);
                printf("\n");
        }
    }
    return 0;
}

图2