传送门
题意
Hamed最近发现了一个字符串T,然后突然就很喜爱这个字符串。他花了一些时间试图找出所有包含T为子串的其他字符串。最后,他变得很累,然后开始考虑下面的问题。 给你一个字符串S,问你有多少种方式提取长度不为0的不重叠的子串,每个子串都必须包含字符串T
分析
字符串匹配问题很自然的可以想到
K
M
P
KMP
KMP算法,因为要求方案数,所以需要
D
P
DP
DP
我们设
f
[
i
]
f[i]
f[i]表示以以第
i
i
i位结尾的方案数,那么就涉及到两种决策
- 不选这一位 那么 f [ i ] = f [ i − 1 ] f[i] = f[i - 1] f[i]=f[i−1]
- 选这一位,那么有两种情况,作为单独的一串连在别的方案的后面,那么就是求匹配的位置,假设第 i i i位的匹配位置为 p [ i ] p[i] p[i],那么方案就是对 1 − p [ i ] − 1 1 - p[i] - 1 1−p[i]−1的方案进行求和,如果是单独的,那么就可以往 1 − p [ i ] 1 - p[i] 1−p[i]进行延长
代码
#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 2e5 + 10;
const ll mod = 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a) {
char c = getchar(); T x = 0, f = 1; while (!isdigit(c)) {if (c == '-')f = -1; c = getchar();}
while (isdigit(c)) {x = (x << 1) + (x << 3) + c - '0'; c = getchar();} a = f * x;
}
int gcd(int a, int b) {return (b > 0) ? gcd(b, a % b) : a;}
char a[N],b[N];
int ne[N],p[N];
ll f[N],sum[N];
int n,m;
void kmp(){
for(int i = 2,j = 0;i <= m;i++){
while(j && b[i] != b[j + 1]) j = ne[j];
if(b[i] == b[j + 1]) j++;
ne[i] = j;
}
}
int main() {
scanf("%s %s",a + 1,b + 1);
n = strlen(a + 1),m = strlen(b + 1);
kmp();
for(int i = 1,j = 0;i <= n;i++){
while(j && a[i] != b[j + 1]) j = ne[j];
if(a[i] == b[j + 1]) j++;
if(j == m){
p[i] = i - m + 1,j = ne[j];
}
}
for(int i = 1;i <= n;i++)
if(!p[i]) p[i] = p[i - 1];
for(int i = 1;i <= n;i++){
f[i] = (f[i - 1] + sum[p[i] - 1] + p[i]) % mod;
sum[i] = (sum[i - 1] + f[i]) % mod;
}
dl(f[n]);
return 0;
}