[SCOI2005]扫雷MINE

题目描述

链接：https://ac.nowcoder.com/acm/problem/20241
来源：牛客网

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

Idea

一道经典的和周围状态相关的DP问题. 但是很注意DP方程的正确性! 调了好久TAT.
可能是比较繁杂的一种做法. 事实上可以枚举上方的格子.

Code

#include <bits/stdc++.h>
#define F(i, a, b) for(int i=a; i<=b; i++)
#define Fd(i, b, a) for(int i=b; i>=a; i--)
#define MAXN 100010

using namespace std;
int n, a[MAXN];
int f[MAXN][2][2][2];
int main(){
    cin>>n;
    F(i, 1, n) cin>>a[i];
    
    f[0][0][0][0] = 1;
    f[0][0][0][1] = 1;
    F(i ,1, n){
        if(a[i]==0) {f[i][0][0][0] = f[i-1][1][0][0]+f[i-1][0][0][0];}
        if(a[i]==1) {
            f[i][1][0][0] = f[i-1][0][1][0]+f[i-1][1][1][0];
            f[i][0][1][0] = f[i-1][1][0][1]+f[i-1][0][0][1];
            f[i][0][0][1] = f[i-1][1][0][0]+f[i-1][0][0][0];
        }else if(a[i]==2){
            f[i][1][1][0] = f[i-1][0][1][1]+f[i-1][1][1][1];
            f[i][1][0][1] = f[i-1][0][1][0]+f[i-1][1][1][0];
            f[i][0][1][1] = f[i-1][0][0][1]+f[i-1][1][0][1];
        }else if(a[i]==3){
            f[i][1][1][1] = f[i-1][0][1][1]+f[i-1][1][1][1];
        }
        
    }
    if(a[n]==0) cout<<f[n][0][0][0]<<endl;
    if(a[n]==1) cout<<f[n][1][0][0]+f[n][0][1][0]<<endl;
    if(a[n]==2) cout<<f[n][1][1][0]<<endl;
    if(a[n]>=3) cout<<"0"<<endl;
    return 0;
    
    
    
}