GalaxyOJ-902 (dp)

最新推荐文章于 2018-08-28 18:51:00 发布

原创最新推荐文章于 2018-08-28 18:51:00 发布 · 828 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

题目专栏收录该内容

165 篇文章

订阅专栏

本文介绍了一道关于一维扫雷游戏的问题，通过动态规划求解给定字符串中问号替换为雷或非雷字符的合法方案数量。讨论了算法实现细节及复杂度分析。

题目

Problem 902: mine
Time Limit: 1000 ms Memory Limit: 256000 KB

Problem Description

有一个 1 维的扫雷游戏，每个格子用表示有雷，用 0/1/2 表示无雷并且相邻格子中有 0/1/2 个雷。给定一个仅包含?、、0、1、2 的字符串 s，问有多少种方法将所有的?改为*/0/1/2 使其合法。

Input

一行一个字符串 s。

Output

一行一个整数表示答案，对 10^9+7 取模。

Sample Input

?1?

Sample Output

2

对于 30%的数据，|S|<=20。对于 60%的数据，|S|<=1000。对于 100%的数据，|S|<=10^6。

Problem Source

雅礼集训10.2

分析

题目不用解释了吧，很明了，数据一百万，目测是 On 复杂度，于是 dp，具体代表的见程序下注释。
开始由于没有取模，一直WA……
复杂度 O(nk)

程序

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
long long n,ans,f[1000005][10];
char a[1000005];

int main(){
    scanf("%s",a);
    n=strlen(a);
    if (a[0]=='*') f[0][4]=f[0][5]=1;
    if (a[0]=='0') f[0][0]=1;
    if (a[0]=='1') f[0][2]=1;
    if (a[0]=='2' || a[n-1]=='2') {puts("0"); return 0;}
    if (a[0]=='?') f[0][0]=f[0][2]=f[0][4]=f[0][5]=1;
    for (int i=1; i<n; i++){
        if (a[i]=='*' || a[i]=='?'){
            f[i][4]=f[i][5]=(f[i-1][2]+f[i-1][3]+f[i-1][4])%(1000000007);
        }
        if (a[i]=='0' || a[i]=='?'){
            f[i][0]=(f[i-1][1]+f[i-1][0])%(1000000007);
        }
        if (a[i]=='1' || a[i]=='?'){
            f[i][2]=(f[i-1][1]+f[i-1][0])%(1000000007);
            f[i][1]=f[i-1][5];
        }
        if (a[i]=='2' || a[i]=='?'){
            f[i][3]=f[i-1][5];
        }
    }
    n--;
    ans=(f[n][0]+f[n][1]+f[n][5])%(1000000007);
    printf("%lld",ans);
}

/*
0   0(下一个不是雷)
1   1(下一个不是雷) 
2   1(下一个是雷) 
3   2(下一个是雷)
4   *(下一个是雷)
5   *(下一个不是雷)
*/