题目描述
相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷,要你根据一些信息找出雷来。 万圣节到了 ,“余”人国流行起了一种简单的扫雷游戏,这个游戏规则和扫雷一样,如果某个格子没有雷,那么它里面的数字 表示和它8连通的格子里面雷的数目。
现在棋盘是n×2的,第一列里面某些格子是雷,而第二列没有雷,如下图: 由于第一列的雷可能有多种方案满足第二列的数的限制,你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。
输入描述:
第一行为N,第二行有N个数,依次为第二列的格子中的数。(1 ≤ N ≤ 10000)
输出描述:
一个数,即第一列中雷的摆放方案数。
示例1
输入 2 1 1 输出 2
关键思路:由题意知道答案一定为0或1或2,由于只有第一个有无地雷需要讨论,所有分类讨论第一个地雷。然后验证方案是否成立。

#include<bits/stdc++.h>
using namespace std;
int a[11000], b[11000];
int ans;
int n;
bool fun(int x) {
    memset(a, 0, sizeof a);
    a[1] = x;
    for (int i = 2; i <= n; i++)
    {
    	//第一列第i个地雷由上面两个地雷和第二列第i-1个的地雷数确定
        a[i] = b[i - 1] - a[i - 1] - a[i - 2];
        if (a[i] != 0 && a[i] != 1) return false;
    }
    //验证最后一个地雷数是否满足
    if(b[n] != a[n] + a[n-1]) return false;
    return true;
}
int main() {

    cin >> n;
    for (int i = 1; i <= n; i++) cin >> b[i];

    if (fun(1)) ans++;
    if (fun(0)) ans++;
    cout << ans;
    return 0;
}