相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷,要你根据一些信息找出雷来。万圣节到了,“余”人国流行起了一种简单的扫雷游戏,这个游戏规则和扫雷一样,如果某个格子没有雷,那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n×2的,第一列里面某些格子是雷,而第二列没有雷,如下图:
由于第一列的雷可能有多种方案满足第二列的数的限制,你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。
一开始想歪了,什么DP、搜索、组合数...其实只要枚举第一个格子的情况,那么就可以推出后面一个格子的情况,之后对于每个格子i,i-1和i的情况都是确定的,递推就好了...
<pre name="code" class="cpp">#include <queue>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int MAXN = 10000 + 100;
int a[MAXN],b[MAXN], n;
int check(int x) { //x只有a[1]是1的时候才有用,表示雷放第一格或第二格
memset(b, -1, sizeof(b)); //1表示要放雷,0表示不放,-1表示还没决定
if (a[1] == 3 || a[n] == 3) return 0; //如果两边是3肯定不合法。
if (a[1] == 2) b[1] = b[2] = 1; //如果边上是2那就放两颗雷
if (a[1] == 1) b[x] = 1, b[3-x] = 0; //1放2不放,或1不放2放
for (int i = 2; i <= n; i ++) {
int cnt = 0;
for (int j = i-1; j <= i; j ++) {
if (b[j] == 1) cnt ++; //因为i-1和i都已经决定了,统计一下有几颗雷
}
if (cnt > a[i]) return 0;//如果大于数字了,不成立
if (cnt == a[i]) b[i+1] = 0; //如果刚好,那么后面一个格子不放
else if (a[i] - cnt > 1 || i == n) return 0; //如果少了两颗以上不成立,因为只有后面一个格子可以补,或者少了,而且在最后一格,因为没格子可以放雷了,也不成立
else b[i+1] = 1; //少了一颗雷,且不是最后一格,那么把后面一个格子放上雷
}
return 1;
}
int main() {
while (scanf("%d",&n) == 1) {
for (int i = 1; i <= n; i ++) scanf("%d",&a[i]);
int res = 0;
if (a[1] == 1) {
res = check(1) + check(2);
} else {
res = check(1);
}
printf("%d\n",res);
}
return 0;
}