【NOIP2015-Day2-T2】洛谷2679:子串 题解

来源:互联网 发布:客户流失数据 编辑:程序博客网 时间:2024/06/05 04:07

显然dp

dp[k][i][j]表示当前匹配了k次,A串匹配到了第i个,B串匹配到了第j个的方案数

一方面,A串的第i个可以不和B串匹配,此时dp[k][i][j]+=dp[k][i-1][j]

另一方面,A串可以选择与B串相同的某长度的后缀与B作为第k次匹配,dp[k][i][j]+=dp[k-1][i-m][j-m],m枚举公共后缀的长度

可以预处理数组far[i][j]表示A串第i个和B串第j个的最长公共后缀

现在空间是炸的,但注意到转移的第一维只会从当前和上一层转移过来,所以可以滚动数组

然而时间也是炸的,但注意到“另一方面”中不必枚举,可以维护一个上一层的前缀和,所以最终复杂度为O(n*m*k)

#include <bits/stdc++.h>#define Pair pair<int,int>#define x first#define y second#define pb push_back#define mp make_pair#define LL long long#define LB long double#define LOWBIT(x) x & (-x)using namespace std;const int MOD=1e9+7;const double eps=1e-10;const int INF=2e9;const int magic=348;int n,m,k;char a[1048],b[248];int far[1048][248];LL dp[2][1048][248],sum[2][1048][248];inline LL mod(LL x){while (x>=MOD) x-=MOD;while (x<0) x+=MOD;return x;}int main (){int i,j,u;scanf("%d%d%d",&n,&m,&k);scanf("%s%s",a+1,b+1);for (i=1;i<=n;i++)for (j=1;j<=m;j++)if (a[i]==b[j]) far[i][j]=far[i-1][j-1]+1; else far[i][j]=0;int p0=0,p1=1;for (i=0;i<=n;i++)dp[p0][i][0]=sum[p0][i][0]=1;for (i=1;i<=n;i++)for (j=1;j<=m;j++)sum[p0][i][j]=mod(sum[p0][i][j]+sum[p0][i-1][j-1]);for (u=1;u<=k;u++){memset(sum[p1],0,sizeof(sum[p1]));for (i=0;i<=n;i++)for (j=0;j<=m;j++){if (!i || !j){dp[p1][i][j]=0;sum[p1][i][j]=0;continue;}dp[p1][i][j]=dp[p1][i-1][j];dp[p1][i][j]=mod(dp[p1][i][j]+sum[p0][i-1][j-1]-((i-far[i][j]-1>=0 && j-far[i][j]-1>=0)?sum[p0][i-far[i][j]-1][j-far[i][j]-1]:0));sum[p1][i][j]=mod(sum[p1][i-1][j-1]+dp[p1][i][j]);}p0^=1;p1^=1;}cout<<dp[p0][n][m]<<endl;return 0;}


原创粉丝点击