J - MUV LUV EXTRA （next数组）

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int N=1e7+100;
int p[N];
int cnt=0;
int ne[N];
ll ans=0;
char s[N];
void getne(int m){
	for(int i=2,j=0;i<=m;i++)
	{
	    while(j&&p[i]!=p[j+1]) j=ne[j];
	    if(p[i]==p[j+1]) j++;
	    ne[i]=j;
	}
}
int main(){ 
	//freopen("in.txt","r",stdin);
	int a,b;scanf("%d%d",&a,&b);
	scanf("%s",s+1);
	int len=strlen(s+1); 
	for(int i=len;i>=1;i--){
		if(s[i]=='.') break;
		p[++cnt]=s[i]-'0';
	}
	getne(cnt);
	ans=(ll)(a-b);
	for(int i=1;i<=cnt;i++){
		ans=max(ans,1ll*a*i-1ll*b*(i-ne[i]));
	}
	printf("%lld\n",ans);
	return 0;
}