bzoj 3620: 似乎在梦中见过的样子 （KMP）

题目描述

传送门

题目大意：给出一个字符串，求所有满足形似于 A+B+A ，且len(A)>=k,len(B)>=1 （位置不同其他性质相同的子串算不同子串,位置相同但拆分不同的子串算同一子串）的子串的数量。

题解

感觉这道题自己的做法时间卡的比较紧。

枚举A+B+A的左端点i，对于[i,n]建立失配函数，从i+k+1开始跑匹配。因为是>=k的，所以直接向后匹配即可。统计答案的时候还需判断一下B是否>=1，如果>=1再更新答案。如果<1，因为我是一直匹配的，所以现在得到的是匹配上的最长长度，这个长度不满足，不代表他的失配也不满足，所以如果<1的话，我们就跳到他的失配，在看看是否可以更新答案，下次匹配从失配开始。需要注意的是位置相同但拆分不同的子串算同一子串，这个限制需要我们对于每个位置维护一个标记，每次计算答案的时候如果匹配到的位置没有标记才ans++.

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 20003
using namespace std;
int t
,n,k,mark
,ti;
char s
,s1
;
void get_fail(int len)
{
t[1]=0; int j;
for (int i=1;i<=len;i++) {
int j=t[i];
while (j!=0&&s1[j]!=s1[i]) j=t[j];
t[i+1]=++j;
}
}
int solve(int head)
{
int pos=head-k-1; ++ti;
int j=head; int i=1; int sum=0;
while (j<=n) {
if (!i||s1[i]==s[j]) i++,j++;
else i=t[i];
if (i>=k+1)
if (j-i+1>=pos+i) {
if (mark[j]!=ti) sum++,mark[j]=ti;
}
else {
i=t[i];
if(i>=k+1) {
if (mark[j]!=ti) sum++,mark[j]=ti;
}
}
}
//cout<<sum<<endl;
return sum;
}
int main()
{
freopen("a.in","r",stdin);
freopen("my.out","w",stdout);
scanf("%s",s+1); n=strlen(s+1);
scanf("%d",&k); int ans=0;
for (int i=1;i<=n-2*k;i++) {
for (int j=1;j<=n-i+1;j++) s1[j]=s[i+j-1];
get_fail(n-i+1);
ans+=solve(i+k+1);
}
printf("%d\n",ans);
}