NOIP2015D2T2 子串动态规划

最新推荐文章于 2019-11-11 13:59:36 发布

Dawn-_-cx

最新推荐文章于 2019-11-11 13:59:36 发布

阅读量143

点赞数

分类专栏： dp NOIP原题

本文链接：https://blog.csdn.net/dhdhdhx/article/details/102508805

版权

dp 同时被 2 个专栏收录

10 篇文章 0 订阅

订阅专栏

NOIP原题

6 篇文章 0 订阅

订阅专栏

题面
这题一打眼
$d p [i] [j] [k]$ 表示方案数啊!
但是仔细想想不对这个 $d p$ 式体现不出当前字符和前一个串的关系 ( $d p$ 式:n你少加一维还怪我了?)
所以我们要再加一维 $0 / 1$ 表示当前字符是接在上一个子串 $/$ 当前字符是一个新的子串的开端
$0 / 1$ 开两个数组更直观一点:
我们用 $a n s [i] [j] [k]$ 表示到 $A$ 字符串第 $i$ 个字符 $B$ 字符串第 $j$ 个字符当前子串数为 $k$ 的答案数
$s u m [i] [j] [k]$ 表示当前在 $A$ 字符串第 $i$ 个字符 $B$ 字符串第 $j$ 个字符当前子串数为 $k$ 时的方案数
然后就是关系式了怎么搞呢
首先考虑当前的 $a n s$ 要么用 $A [i]$ 要么不用
所以 $a n s [i] [j] [l] = a n s [i - 1] [j] [l] + s u m [i] [j] [l]$
显而易见当 $A[i]\neq B[i]$ 时 $s u m$ 的值就是 $0$
经过思考可知当 $A [i] = B [i]$ 时此时要么接在上个子串上要么开一个新的子串
所以 $s u m [i] [j] [l] = s u m [i - 1] [j - 1] [l] + a n s [i - 1] [j - 1] [l - 1]$
哈哈做完了？想得美
$d p [1000] [200] [200] [2] ? ?$ 显然是炸的所以我们要优化空间
注意到第一维貌似没什么卵用所以直接滚掉
空间复杂度 $O (4 m k)$ 时间复杂度 $O (n m k)$ 没啥大问题
上代码

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
int n,m,k,now,ans[2][202][202],sum[2][202][202];
char a[1010],b[202];
int main(){
    cin>>n>>m>>k;
    scanf("%s%s",a+1,b+1);
    ans[0][0][0]=ans[1][0][0]=now=1;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            for(int l=1;l<=k;l++){
                if(a[i]==b[j])sum[now][j][l]=(sum[now^1][j-1][l]+ans[now^1][j-1][l-1])%mod;
                else sum[now][j][l]=0;
                ans[now][j][l]=(ans[now^1][j][l]+sum[now][j][l])%mod;
            }
        }
        now^=1;
    }
    printf("%d\n",ans[now^1][m][k]);
}