[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;
}