取子串

取子串

思路

刚开始不怎么敢写DP
因为这题的状态转移看起来还是比较麻烦的

其实当我们找到一个T串的时候
可以简单分成两种情况

第一种情况：

只存在一个子串
那么存在的这个字串必定包含找到的这个串 
将这个T串往前延伸，可以得到i-l2+1种方案

第二种情况：

存在多个字串
这样的话就需要借助前面的结果转移
可以枚举当前子串延伸的长度
再统计当前延伸的边界左边的方案数

我们不妨定义三个数组
dp[i] => 以第i个字母结尾的方案数
tmp[i] => 前i个字母组成的方案数

第一种情况的结果可以直接下标相减得到
第二种 dp[i]的增加量其实就是tmp[i]到tmp[i-l2]的和

显然我们再定义个前缀和数组sum
sum[i] => tmp[1]到tmp[i]的和

处理完T串的情况，那么如果当前字母无法和之前的字母构成合法子串
就可以直接赋值 dp[i]=dp[i-1]
因为以当前字母结尾的方案一定是以上一个字母结尾的方案的末尾加上当期字母
说起来好像有点绕，其实还是比较显然

这样就能完成dp数组的转移了
但是这样按数据范围只有60分（但数据太水可以过）

这样做复杂度是n^2的
可以发现状态转移已经O(1)了
而复杂度其实都堆在匹配T串上

那么就可以用Hash来处理字符串
实现O(1)的匹配

算法时间复杂度：O(n)

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

#define FOR(i,a,b) for(int i=(a),i##_end_=(b);i<=i##_end_;++i)
#define DOR(i,a,b) for(int i=(a),i##_end_=(b);i>=i##_end_;--i)
#define LL long long
#define INF 0x3f3f3f3f
#define M 100005
#define N 10
#define P 1000000007
#define B 233

void chkmx(int &a,int b) {if(a<b)a=b;}
void chkmi(int &a,int b) {if(a>b)a=b;}
int MAX(int a,int b) {return a>b?a:b;}
int MIN(int a,int b) {return a<b?a:b;}
void Add(LL &a,LL b) {a=(a+b)%P;}

char S[M],T[M];
LL dp[M],tmp[M],sum[M];
int l1,l2;
struct Hash{
    unsigned int H[M],Base[M],HashT;
    void Init() {
        FOR(i,1,l2) HashT=HashT*B+T[i];
        Base[0]=1;
        FOR(i,1,l1) Base[i]=Base[i-1]*B;
        FOR(i,1,l1) H[i]=H[i-1]*B+S[i];
    }
    bool check(int s) {
        return H[s]-H[s-l2]*Base[l2]==HashT;
    }
}Hash;

int main() {
//  freopen("substr.in","r",stdin);
//  freopen("substr.out","w",stdout);
    scanf("%s%s",S+1,T+1);
    l1=strlen(S+1),l2=strlen(T+1);
    Hash.Init();
    FOR(i,l2,l1) {
        if(Hash.check(i)) Add(dp[i],sum[i-l2]+i-l2+1);
        else dp[i]=dp[i-1];
        Add(tmp[i],tmp[i-1]+dp[i]);
        Add(sum[i],sum[i-1]+tmp[i]);
    }
    cout<<tmp[l1]<<endl;
    return 0;
}