问题 D: 扫雷

题目描述

有一种简单的扫雷游戏： 在 n 行 2 列的方格棋盘上， 左列某些方格内埋有地雷，而右列每个方格中都有一个数字（ 0～3）， 第 I 格的数字表示： 左列第 I－1、 I、 I＋1格（ 即： 上、 中、 下三格） 中埋雷的总数。 如下所示： 左图是初始状态， 右图是扫雷完成状态（ 插小旗的方格内有雷）。

你的任务是： 根据右列的数字分析出左列格子中的地雷（ 0 表示无雷， 1 表示有
雷）， 并且统计出左列格子中地雷的总数。

输入

第一行， 一个整数 N（ 3≤N≤40）， 第二行有 N 个数字（ 以空格相隔）， 表示右列格子中的数字。

输出

第一行是 N 个 0、 1 数字（ 没有空格相隔）， 表示左列每格中有无地雷。 第二行一个整数， 表示地雷总数。
题目保证一定有唯一解。 

样例输入 

7 
1 2 3 2 2 2 2

样例输出 

0111011
5

 我们可以假设第一个有雷 ，那么就可以推出后面是否有雷(doit函数)，如果假设成立，exit结束程序，并输出；如果假设不成立，那么就说明第一个有雷，再推一遍。

接下来放代码：

#include<bits/stdc++.h>
using namespace std;
int a[50],b[50],n;
void doit()
{
    int s=a[1];
    for(int i=2;i<=n;i++)
    {
        a[i]=b[i-1]-a[i-1]-a[i-2];
        if(a[i]<0||a[i]>1)
            return ;
        s=s+a[i];//s统计地雷个数
    }
    if(b[n]!=a[n-1]+a[n])
        return ;
    for(int i=1;i<=n;i++)
        cout<<a[i];
    cout<<endl<<s<<endl;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>b[i];
    a[1]=0;
    doit();
    a[1]=1;
    doit();
    return 0;
}