2018.09.18 atcoder Best Representation（kmp）

最新推荐文章于 2024-04-12 04:51:17 发布

SC.ldxcaicai

最新推荐文章于 2024-04-12 04:51:17 发布

阅读量482

点赞数 2

CC 4.0 BY-SA版权

分类专栏： # kmp 文章标签：字符串

本文链接：https://blog.youkuaiyun.com/dreaming__ldx/article/details/82765159

kmp 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用KMP算法解决特定字符串划分问题的方法。该问题要求将字符串划分为最少数量的子串，使得每个子串都是自身的一个循环节。通过KMP算法预处理前后缀匹配情况，判断是否存在合适的划分方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门
思路简单不知为何调试了很久。
显然要么分成n个（所有字符相同），要么分成1个（原字符串无循环节），要么分成两个（有长度至少为2的循环节）。
一开始以为可以直接hash搞定。
后来wa了几次之后发现可以轻松举出反例于是弃了hash。
kmp大法好啊，判完循环节之后直接枚举两个子串的断点判是不是前缀与后缀同时满足条件就行了。
代码：

#include<bits/stdc++.h>
#define N 500005
#define mod 1000000007
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
char s[N],t[N];
int n,flag,pre[N],suf[N],ans=0;
void getfail(int *fail,char *s)
{
	fail[0]=-1;
	int j=0;
	for(int i=1;i<n;++i){
		while(j&&s[i]!=s[j])j=fail[j];
		if(s[i]==s[j])fail[i+1]=++j;
		else fail[i+1]=0;
	}
}
inline bool checkpre(int pos){return pre[pos]==0||pos%(pos-pre[pos]);}
inline bool checksuf(int pos){return suf[pos]==0||pos%(pos-suf[pos]);}
int main(){
	scanf("%s",s),n=strlen(s);
	for(int i=0;i<n;++i)t[i]=s[n-i-1];
	getfail(pre,s),getfail(suf,t),t[n]='\0';
	flag=checkpre(n)?1:2;
	bool f=true;
	for(int i=1;i<n;++i)if(s[i]!=s[i-1]){f=false;break;}
	if(f)flag=n;
	cout<<flag<<'\n';
	if(flag!=2){cout<<1;return 0;}
	for(int i=0;i<n;++i){
		int tmp=n-i-1;
		if(checkpre(i+1)&&checksuf(tmp))++ans;
	}
	cout<<ans;
	return 0;
}