【BZOJ】1088 [SCOI2005] 扫雷Mine

Description

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

Input

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

Output

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

Solution

枚举前2个位置雷的摆法，后面的状态是确定的（可以递推）。

#include<stdio.h>

int a[10005],b[10005],ans,n;

inline void solve(const int &a1,const int &a2)
{
    if (a[1]!=a1+a2) return;
    b[1]=a1;b[2]=a2;
    for (int i=3;i<=n;i++) if((b[i]=a[i-1]-b[i-1]-b[i-2])<0) return;
    if (a[n]==b[n]+b[n-1]) ans++;
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",a+i);
    if (n==1){puts("1");return 0;}
    if (n==2)
    {
        if (a[1]+a[2]==0 || a[1]+a[2]==4){puts("1");return 0;}
        if (a[1]+a[2]!=2) {puts("0");return 0;}
        puts("2");
        return 0;
    }
    solve(0,0);solve(1,0);solve(0,1);solve(1,1);
    printf("%d",ans);
}