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

最新推荐文章于 2022-10-24 20:11:57 发布

原创最新推荐文章于 2022-10-24 20:11:57 发布 · 168 阅读

0 ·

CC 4.0 BY-SA版权

ybt 同时被 2 个专栏收录

91 篇文章

订阅专栏

kmp

4 篇文章

订阅专栏

本文介绍了一种子串拆分算法，通过枚举子串并计算next数组来判断子串是否能被拆分为ABAABAABA形式。适用于字符串长度不大于15000的情况。

子串拆分

题目链接：子串拆分

题目大意

在这里插入图片描述

解题思路

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

我们枚举子串，求这个子串的 $n e x t$ 数组，而且只要两个串不重合且长度大于 $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;
}