【SCOI2005】扫雷

bzoj1088

Description

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

Input

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

Output

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

Sample Input

2

1 1

Sample Output

2

大概也是大水题系列了…因为地图只有两行，所以定下一个格子，所有的格子就都定下来了，所以答案只有可能是{0，1，2}…这时候我们要分情况讨论。

①当第一个格子是1时：
1. 第一个格子有雷。
2. 第二个格子有雷。
②当第一个格子是2时：
第一二个格子都是雷。
③当第一个格子是0时：
第一二个格子都无雷。

其他的则可以通过递推式推出，验证合法即可。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int size=10010;
int num[size],n;
int f[size];

bool full()
{
    for(int i=2;i<=n;i++)
    {
        f[i+1]=num[i]-f[i]-f[i-1];
    }
    if(f[n+1]) return false;
    return true;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&num[i]);
    }
    int ans=0;
    if(num[1]==0)
    {
        if(full()) ans++;
    }
    else if(num[1]==1)
    {
        f[1]=1;
        if(full()) ans++;
        memset(f,0,sizeof(f));
        f[2]=1;
        if(full()) ans++;
    }
    else 
    {
        f[1]=f[2]=1;
        if(full()) ans++;       
    }
    printf("%d",ans);
    return 0;
}