BZOJ_P1088&Codevs_P2452 [SCOI2005]扫雷(DP)

2005年省队选拔赛四川
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 大师 Master
题目描述 Description
相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了，“余”人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷，由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。

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

输出描述 Output Description
一个数，即第一列中雷的摆放方案数。

样例输入 Sample Input
2
1 1

样例输出 Sample Output
2

数据范围及提示 Data Size & Hint
见题面

首尾特判，中间方案数考虑前一个的尾部方案数，
代码有点长

#include<cstdio>
#include<iostream>
using namespace std;
#define N 10005
int n,a[N],f[N][4],ans;
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    switch(a[1]){
        case 0:
            f[1][0]=1;break;
        case 1:
            f[1][1]=1,f[1][2]=1;break;
        case 2:
            f[1][3]=1;break;
    }
    for(int i=2;i<n;i++){
        switch(a[i]){
            case 0:
                f[i][0]=f[i-1][0];break;
            case 1:
                f[i][0]=f[i-1][2],f[i][1]=f[i-1][0],f[i][2]=f[i-1][1];break;
            case 2:
                f[i][2]=f[i-1][3],f[i][1]=f[i-1][2],f[i][3]=f[i-1][1];break;
            case 3:
                f[i][3]=f[i-1][3];break;
        }
    }
    switch(a[n]){
        case 0:
            ans=f[n-1][0];break;
        case 1:
            ans=f[n-1][2]+f[n-1][1];break;
        case 2:
            ans=f[n-1][3];break;
    }
    printf("%d",ans);
    return 0;
}