【ybtoj 高效进阶 2.3】D.子串拆分【KMP】

最新推荐文章于 2024-07-17 17:02:45 发布

ssllyr

最新推荐文章于 2024-07-17 17:02:45 发布

阅读量147

点赞数

分类专栏： KMP YbtOJ

本文链接：https://blog.csdn.net/dglyr/article/details/118595114

版权

YbtOJ 同时被 2 个专栏收录

69 篇文章 0 订阅

订阅专栏

KMP

4 篇文章 0 订阅

订阅专栏

在这里插入图片描述
题目链接：子串拆分

分析

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

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
char s[15001],s1[15001];
int kk,n,ans;
int next[15001];
int main()
{
	scanf("%s",s1+1);
	cin>>kk;
	n=strlen(s1+1);
	for(int i=1;i<=n;i++)
	{
		memset(s,0,sizeof(s));
		for(int j=1;j+i-1<=n;j++)
		{
			s[j]=s1[j+i-1];
		}
		int wx=strlen(s+1);
		memset(next,0,sizeof(next));
		int j=0;
		for(int k=2;k<=wx;k++)
		{
			while(j&&s[k]!=s[j+1])j=next[j];
			if(s[k]==s[j+1]) j++;
			next[k]=j;
		}
		j=0;
		for(int k=1;k<=wx;k++)
		{
			while(j&&s[k]!=s[j+1])j=next[j];
			if(s[k]==s[j+1]) j++;
			while(j*2>=k) j=next[j];
			if(j>=kk) ans++;
		}
	}
	cout<<ans;
	return 0;
 }