【ybt】【字符串 kmp 课过 例4】子串拆分

子串拆分

题目链接：子串拆分

---

题目大意

解题思路

题目要求我们将子串拆分成 $ABA$ 的形式。

我们枚举子串，求这个子串的 $next$ 数组，而且只要两个串不重合且长度大于 $k$ 就可以计入答案。

code

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

char ss[15010],s[15010];
int K,n,ans;
int nxt[15010];

int main()
{
	scanf("%s",ss+1);
	scanf("%d",&K);
	n=strlen(ss+1);
	for(int i=1;i<=n;i++)
	{
		memset(s,0,sizeof(s));
		for(int j=1;j+i-1<=n;j++)
			s[j]=ss[j+i-1];
		int nn=strlen(s+1);
		memset(nxt,0,sizeof(nxt));
		int j=0;
		for(int k=2;k<=nn;k++)
		{
			while(j&&s[k]!=s[j+1]) j=nxt[j];
			if(s[k]==s[j+1]) j++;
			nxt[k]=j;
		}
		j=0;
		for(int k=1;k<=nn;k++)
		{
			while(j&&s[k]!=s[j+1]) j=nxt[j];
			if(s[k]==s[j+1]) j++;
			while(j*2>=k) j=nxt[j];
			if(j>=K) ans++;
		}
	}
	cout<<ans<<endl;
}