VJP1193 扫雷(状压）

链接

保存当前行和前一行两行的状态 

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define N 10010
int a[N],dp[4][4][N],d[4]={0,1,1,2};
int main()
{
    int i,j,k,n,g;
    while(scanf("%d",&n)!=EOF)
    {
        memset(dp,0,sizeof(dp));
        for(i = 1; i <= n ; i++)
        scanf("%d",&a[i]);
        if(n==1)
        {
            printf("1
");
            continue;
        }
        for(i = 0 ; i <= 1 ; i++)
        {
            for(j = 0; j <= 1 ; j++)
            if(i+j==a[1])
            dp[i][j][2] = 1;
        }
        for(i =3 ; i <= n ; i++)
        {
            for(j = 0 ; j < 2 ; j++)
            {
                if(j>a[i]||j>a[i-1])
                continue;
                for(g = 0 ; g < 2 ; g++)
                {
                    if(j+g>a[i]||j+g>a[i-1])
                    continue;
                    for(int e = 0 ; e < 2 ; e++)
                    {
                        if(j+g+e==a[i-1])
                        dp[j][g][i] += dp[g][e][i-1];
                    }
                }
            }
        }
        int ans=0;
        for(i = 0 ; i < 2 ; i++)
            for(j = 0 ; j < 2 ; j++)
            if(i+j==a[n])
            ans+=dp[i][j][n];
        printf("%d
",ans);
    }
    return 0;
}