每日算法题（Day7）----扫雷

题目描述

相信大家都玩过扫雷的游戏。那是在一个n×m的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了，“余”人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷。
由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。

输入格式

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

输出格式

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

分析

依旧是dp问题
分第一个格子放与不放两大类讨论，结果只有0，1，2三种情况
0.33^10，碰运气ac也不是不可能

样例

2
1 1
------------------------
2

代码

#include<cstdio>
#include<cstring>
#include<iostream>using namespace std;int n,a[20010],b[20010];
bool judge()
{
    for(int i=1;i<=n;i++)if(a[i]>3) return false;if(a[1]>2||a[n]>2) return false;return true;
}
int ans(int k)
{
    b[1]=k;for(int i=1;i<=n;i++){
    b[i+1]=a[i]-b[i-1]-b[i];}b[0]=0;b[n+1]=0;for(int i=1;i<=n;i++)if(b[i]<0||b[i]>1) return 0;for(int i=1;i<=n;i++){
    if(b[i-1]+b[i]+b[i+1]!=a[i]) return 0;}return 1;
}
int main()
{
    scanf("%d",&n);for(int i=1;i<=n;i++){
    scanf("%d",&a[i]);}if(!judge()){
    cout<<"0";return 0;}cout<<ans(1)+ans(0)<<endl;return 0;
}