ARC 060F Best Representation KMP(循環節)

題意:定義一個字符串合法:當該字符串沒有循環節存在,
給出string s,令F=(f1,f2..fm) 滿足fi爲s的某一部分.f1,f2,..fm連起來爲s.並且任意fi爲合法
|s|<=5e5,求出F表示中最小的m.並求出最小m的方案數?

求最小m 利用next求出最小循環節,若len-fail[len]=1 則m=len,若len%(len-f[len])!=0 則m=1

其餘有循環節情況,都可以按照前len-1個分一組,最後一個字符分一組 則m=2.
求m=2時的方案數 也就是求有多少個i滿足  前綴i合法&&後綴i+1合法.

正着和逆着求一次fail函數,即可O(N)判斷前後綴是否(合法)包含循環節.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e6+20;
const ll mod=1e9+7;
char s[N],t[N];
int fail[N],suf[N];
void getfail(int *fail,char *s)
{
	fail[0]=-1;
	int i=0,j=-1;
	while(s[i])
	{
		if(j==-1||s[i]==s[j])
		{
			fail[i+1]=j+1;
			i++,j++;
		}
		else
			j=fail[j];
	}
}
int main()
{
	while(scanf("%s",s)!=EOF)
	{
		ll ans=0;
		int len=strlen(s);
		for(int i=0;s[i];i++)
			t[i]=s[len-i-1];
		t[len]='\0';
		getfail(fail,s);
		getfail(suf,t);
		int m;
	//	cout<<fail[len]<<endl;
		if(fail[len]==0||len%(len-fail[len]))
			m=1;
		else
			m=2;
		bool flag=true;
		for(int i=1;s[i];i++)
		{
			if(s[i]!=s[i-1])
				flag=false;
		}	
		if(flag)
			m=len;
		if(m==1||flag)
			ans=1;
		else
		{
			for(int i=0;i<len-1;i++)
			{
				int k=len-i-1;
			//	cout<<i<<' '<<k<<endl;
				if((fail[i+1]==0||(i+1)%((i+1)-fail[i+1])) &&(suf[k]==0||(k)%(k-suf[k])))
					ans++;
			}
		}	
		cout<<m<<endl<<ans<<endl;
	}
	return 0;
}

關於循環節:

若包含循環節 i-fail[i]爲最小循環節 不存在更小 否則和fail函數矛盾