扫雷MINE——枚举+递推

题目描述

相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷，要你根据一些信息找出雷来。

万圣节到了 ，“余”人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字 表示和它8连通的格子里面雷的数目。

现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷，如下图： 由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。

输入描述

第一行为N，第二行有N个数，依次为第二列的格子中的数。（1 ≤ N ≤ 10000）

输出描述

一个数，即第一列中雷的摆放方案数。

示例1

输入

2

1 1

输出

2

思路

• 两列数据，第一列摆放雷，第二列没有雷，第二列中的数表示与它8联通的格子里面雷的数量
• 根据map[0]的值，我们可以推断出mine[0]和mine[1]的值

• 根据扫雷的游戏规则，我们可以推断出以下公式：

map[i] = mine[i-1] + mine[i] + mine[i+1]

mine[i] = map[i-1] - mine[i-2] - mine[i-1]

• 知道了mine[0]和mine[1]就可以推断出mine[2]以及后面的值
• 推断出所有的mine之后，再判断mine值是否合法

1. mine[i] 只能为0或1
2. mine[n-1] + mine[n-2] == map[n-1]

代码实现

#include <iostream>

using namespace std;

const int N = 1e4 + 10;

int n;
int mine[N], map[N];	// mine表示第一列，map表示第二列

int solve()
{
    // map[i] = mine[i-1] + mine[i] + mine[i+1]
    // mine[i] = map[i-1] - mine[i-2] - mine[i-1]
    for(int i = 2; i< n;i ++)	// 从下标为2开始，推出每一个mine的值，并判断是否合法
    {
        mine[i] = map[i-1] - mine[i-2] - mine[i-1];
        if(mine[i] < 0 || mine[i] > 1) return 0;
    }
    if(mine[n-2] + mine[n-1] != map[n-1]) return 0;	// 最后两项是否合法
    return 1;	// 这种方案成立
}

int main()
{
    cin >> n;
    for(int i = 0;i < n;i ++) cin >> map[i];
    if(n == 1)    // 只有一行，一种方案数
    {
        cout << 1 << endl;
        return 0;
    }
    
    if(map[0] == 2)	// 第一行、第二行一共两颗雷，每行一个
    {
        mine[0] = 1, mine[1] = 1;
        cout << solve() << endl;
    }
    else if(map[0] == 1)	// 第一行、第二行一共一颗雷，分两种情况
    {
        mine[0] = 1;	// 第一行有雷，第二行没有
        mine[1] = 0;
        int t1 = solve();
        mine[0] = 0;	// 第二行有雷， 第一行没有
        mine[1] = 1;
        int t2 = solve();
        cout << t1 + t2 << endl;	// 将两种情况得到的方案数相加，为总的方案数
    }
    else 	// 第一行、第二行一颗雷也没有
    {
        cout << solve() << endl;
    }
    
    return 0;
}