404D Minesweeper 1D(记忆化搜寻,dp)
404D Minesweeper 1D(记忆化搜索,dp)
题目链接:http://codeforces.com/problemset/problem/404/D
题意:扫雷的游戏,给一串字符,'*'代表此处有地雷,'?'代表未知,'0'代表左右有0个,'1'代表左右有1个,'2'代表左右有2个;为给的字符串共有多少种合法的地雷分步情况;
解法:dp;ans[i][pre][now]表示i的位置之前一个位置地雷状态是pre(0/1),i位置地雷状态是now(0/1).在记忆化搜索的时候,只需排除非法,搜索合法的状态就可以;
代码:
/**************************************************** * author:xiefubao *******************************************************/ #pragma comment(linker, "/STACK:102400000,102400000") #include <iostream> #include <cstring> #include <cstdlib> #include <cstdio> #include <queue> #include <vector> #include <algorithm> #include <cmath> #include <map> #include <set> #include <stack> #include <string.h> using namespace std; #define eps 1e-8 typedef long long LL; int inf=1000000007 ; int ans[1000100][2][2]; char s[1000100]; int len; int dfs(int add,int pre,int now) { if(add==len) return now==0; if(ans[add][pre][now]!=-1) return ans[add][pre][now]; if(now==1&&s[add]!='*'&&s[add]!='?') return 0; if(now==0&&s[add]=='*') return 0; int tool=0; for(int i=0;i<=1;i++) { int next=i; if(s[add]=='0'&&(pre+next)!=0) continue; if(s[add]=='1'&&(pre+next)!=1) continue; if(s[add]=='2'&&(pre+next)!=2) continue; tool+=dfs(add+1,now,next);tool%=inf; } return ans[add][pre][now]=tool; } int main() { scanf("%s",s); len=strlen(s); memset(ans,-1,sizeof ans); cout<<(dfs(0,0,0)+dfs(0,0,1))%inf<<'\n'; return 0; }