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],n,k,mark[N],ti;
char s[N],s1[N];
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);
}