NOIP2015子串

题目来源：https://www.luogu.org/problem/show?pid=2679

很难想的dp，看了题解后才理清思路。

最先想到的是定义f[i][j][k]为a串匹配到i处，b串匹配到j处，取了k次的方案数。

f[i][j][k]=f[i-1][j][k]+(f[i-1][j-1][k-1]+...+f[i-len][j-len][k-1])，其中a串从i-len到i与b串j-len到j相同。

这样复杂度为O(n^2mk)，显然会超时。

分别定义数组f[i][j][k][1]、f[i][j][k][0]分别为使用a[i]==b[j]和未使用的方案数（首先保证a[i]是否等于b[j]）。

这样f[i][j][k][0]=f[i-1][j][k][0]+f[i-1][j][k][1]。

然而a串和b串匹配到i、j处时相同的区域可能是连续的。

因此f[i][j][k][1]=f[i-1][j-1][k-1][1]+f[i-1][j-1][k-1][0]+f[i-1][j-1][k][1](a[i]==b[j])，

其中f[i-1][j-1][k-1][1]表示即使用到a[i]==b[j]，又使用到a[i-1]==b[j-1]，但取的是不同的两次。

f[i-1][j-1][k-1][0]表示只使用到a[i]==b[j]。

f[i-1][j-1][k][1]表示即使用到a[i]==b[j]，又使用到a[i-1]==b[j-1]，且二者在取的同一次中。

故状态转移方程为：

f[i][j][k][0]=f[i-1][j][k][0]+f[i-1][j][k][1]

f[i][j][k][1]=f[i-1][j-1][k-1][1]+f[i-1][j-1][k-1][0]+f[i-1][j-1][k][1](a[i]==b[j])

时间复杂度O(nmk)。

然而数组f的大小为1000*200*200*2=8e7，根本开不下。

故需要对空间进行优化。

使用滚动数组的方法来让f减小一维。

观察状态转移方程，我们可以看到f[i]只与f[i-1]有关，也就是说f[i]只能从f[i-1]转移过来。

因此f[i]只需开到f[2]即可存下。

转移时f[i%2]从f[(i-1)%2]转移过来。

考虑f数组的初始化，f应先清零，并令f[i%2][1][1][1]=1。

最后，运算过程中不要忘了取模。

代码：

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
using namespace std;
const int mod=1e9+7;
int f[2][202][202][2];
char a[1001],b[1001];
int main()
{
    ios::sync_with_stdio(false);
    int n,m,l;cin>>n>>m>>l;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int j=1;j<=m;j++)cin>>b[j];
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            for(int k=1;k<=l;k++)
            {
                int t=i%2,c=(t+1)%2;
                f[t][j][k][0]=f[t][j][k][1]=0;
                f[t][j][k][0]=f[c][j][k][0]+f[c][j][k][1];
                f[t][j][k][0]%=mod;
                if(a[i]==b[j])
                {
                    f[t][j][k][1]=((f[c][j-1][k-1][0]+f[c][j-1][k-1][1])%mod+f[c][j-1][k][1])%mod;
                    if(k==1&&j==1&&f[t][j][k][1]==0)f[t][j][k][1]=1;
                }
            }
        }
    }
    cout<<(f[n%2][m][l][1]+f[n%2][m][l][0])%mod;
    return 0;
}