[BZOJ1088][SCOI2005]扫雷Mine（dp）

题目描述

传送门

题目大意：第一行有雷，第二行没有雷，但是有数字。问满足这个数字的第一行的方案数。

题解

f(i,s)表示满足前i个格子的限制，并且第一行最后三个格子的状态是s的方案数
直接dp就可以了

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 10005

int n,a[N];
long long ans,f[N][10];

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;++i) scanf("%d",&a[i]);
    for (int i=0;i<=7;++i)
        if ((i>>2&1)==0&&(i>>1&1)+(i&1)==a[1])
            ++f[1][i];
    for (int i=1;i<n;++i)
        for (int j=0;j<=7;++j)
            for (int k=0;k<=1;++k)
            {
                int s=(j&3)<<1|k;
                if ((s>>2&1)+(s>>1&1)+(s&1)==a[i+1])
                    f[i+1][s]+=f[i][j];
            }
    for (int i=0;i<=7;++i)
        if ((i&1)==0) ans+=f[n][i];
    printf("%lld\n",ans);
}