BZOJ1088: [SCOI2005]扫雷Mine

n<=10000，n*2的格子玩扫雷，给下面一行的数字，问上面一行有几种方案。

？？？？我写的DP，貌似有更简洁的写法。。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
//#include<iostream>
//#include<time.h>
using namespace std;

int n;
#define maxn 10011
int f[maxn][5][5],a[maxn];
int main()
{
    scanf("%d",&n);
    int tmp=0;
    for (int i=1;i<=n;i++) scanf("%d",&a[i]),tmp=max(tmp,a[i]);
    if (tmp>=4) {puts("0");return 0;}
    if (n==1) {if (a[1]==1) puts("1");else puts("0");return 0;}
    f[1][a[2]][a[1]]=1;
    if (a[2]>0 && a[1]>0) f[1][a[2]-1][a[1]-1]=1;
    a[n+1]=3;
    for (int i=2;i<=n;i++)
        for (int j=0;j<=a[i];j++)
        {
            f[i][a[i+1]][j]=f[i-1][j][0];
            if (a[i+1]>0) f[i][a[i+1]-1][j]=f[i-1][j+1][1];
        }
//    for (int i=1;i<=n;i++)
//        for (int j=0;j<=4;j++)
//            for (int k=0;k<=4;k++)
//                cout<<i<<' '<<j<<' '<<k<<' '<<f[i][j][k]<<endl;
    printf("%d\n",f[n][a[n+1]-1][0]+f[n][a[n+1]][0]);
    return 0;
}