P2327 [SCOI2005] 扫雷

题目传送门 

题目描述

相信大家都玩过扫雷的游戏。那是在一个n×m的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了，“余”人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷，如下图：

由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。

输入格式

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

输出格式

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

输入输出样例

输入 #1

2
1  1

输出 #1

2

思路 

 用bool函数标记，再累加，bool函数内用递推式求b[i+1]=a[i]-b[i-1]-b[i];

#include<bits/stdc++.h>
using namespace std;
int n,a[10009],b[10009];
bool w(int i)
{
    b[0]=0;
    for(i=1;i<=n;i++)
    {
        b[i+1]=a[i]-b[i-1]-b[i];
        if(b[i+1]>1||b[i]<0)return 0;
    }
    if(b[n+1]==0)return 1;
    return 0;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    int s=0;
    for(b[1]=0;b[1]<2;b[1]++)s+=w(1);
    cout<<s;
    return 0;
}

 如有错误，欢迎大家评论区指出！感谢！