BZOJ_1009 GT考试

zhu8655

于 2015-06-02 16:31:45 发布

阅读量346

点赞数

分类专栏：矩阵乘法 KMP 动态规划

本文链接：https://blog.csdn.net/zhu8655/article/details/46331753

版权

动态规划同时被 3 个专栏收录

2 篇文章 0 订阅

订阅专栏

矩阵乘法

1 篇文章 0 订阅

订阅专栏

KMP

1 篇文章 0 订阅

订阅专栏

1.题目相关

标签：DP 矩阵乘法 KMP
题目地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1009
题目大意：中文题。

2.思路

比较基础的矩阵乘法加速动态规划。
令 $f[i][j]$ 表示到 $i$ 位，且当前后缀与不吉利串匹配到第 $j$ 位的方案数。
$f [i] [j] = \sum k = 0 m - 1 f [i - 1] [k] \cdot p r e [k] [j]$ $f[i][j] = \sum_{k = 0}^{m - 1} f[i-1][k] \cdot pre[k][j]$
其中 $pre[k][j]$ 表示从匹配 $k$ 位的串加一个字符后变为匹配了 $j$ 位的方案数。
但它的时间复杂度为 $O(nm)$ 难以接受，所以考虑优化。
可以观察到转移方程是一个线性齐次递推式，所以可以用矩阵乘法来加速。
至于构造这个矩阵可以用 KMP 来进行。

void prep() {
    int j = 0;
    for (int i = 2; i <= m; i++) {
        while (j > 0 && a[j+1] != a[i]) j = next[j];
        if (a[j+1] == a[i]) j++;
        next[i] = j;
    }
    for (int i = 0; i < m; i++) {
        for (int j = 0; j <= 9; j++) {
            int k = i;
            while (k > 0 && a[k+1] != j) k = next[k];
            if (a[k+1] == j) k++;
            if (k < m) pre[i][k] = (pre[i][k]+1)%MOD;   //由i位转移到k位的方案++.
        }
    }
    mt[0][0] = 1;   //f[0][0]的方案数为1.
}