題意: 求字符串 s[i…len−1] and s[0…len−1] i>0 最長公共前綴長度求解過程的比較次數
題解: 當時一看就直接AC自動機然後用fail指針就行,結果輸入字符是ASCII那就超內存了。用擴展kmp求出每一個後綴的最長公共前綴求和即可。注意 當前下標加上最長前綴長度超過字符串長 ans--。
因爲是求比較次數。
擴展KMP解決的問題:
定義母串S和子串T,S的長度爲n,T的長度爲m;
求 字符串T 與 字符串S的每一個後綴 的最長公共前綴;
也就是說,設有extend數組:extend[i]表示T與S[i,n-1]的最長公共前綴,要求出所有extend[i](0<=i<n)。
設有輔助數組next[i]表示T[i,m-1]和T的最長公共前綴長度
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int K=1000005;
int nt[K],extand[K];
char S[K],T[K];
int a[K];
ll ans;
void Getnext(char *T,int *next)
{
int len=strlen(T),a=0;
next[0]=len;
while(a<len-1 && T[a]==T[a+1]) a++;
next[1]=a;
a=1;
for(int k=2; k<len; k++)
{
int p=a+next[a]-1,L=next[k-a];
if( (k-1)+L >= p)
{
int j = (p-k+1)>0 ? (p-k+1) : 0;
while(k+j<len && T[k+j]==T[j]) j++;
next[k]=j;
a=k;
}
else
next[k]=L;
}
}
void GetExtand(char *S,char *T,int *next)
{
Getnext(T,next);
int slen=strlen(S),tlen=strlen(T),a=0;
int MinLen = slen < tlen ? slen : tlen;
while(a<MinLen && S[a]==T[a]) a++;
extand[0]=a;
a=0;
for(int k=1; k<slen; k++)
{
int p=a+extand[a]-1, L=next[k-a];
if( (k-1)+L >= p)
{
int j= (p-k+1) > 0 ? (p-k+1) : 0;
while(k+j<slen && j<tlen && S[k+j]==T[j]) j++;
extand[k]=j;
a=k;
}
else
extand[k]=L;
}
}
//以上是模版
int main(void)
{
int tl;scanf("%lld",&tl);
while(tl--)
{
memset(nt,0,sizeof(nt));
memset(extand,0,sizeof(extand));
memset(a,0,sizeof(a));
scanf("%s",S);
ans = 0;
strcpy(T,S);
GetExtand(S,T,nt);
int len = strlen(T);
queue<int> id;
ans = 0;
ans+=len-1;
for(int i=1; i<len; i++) {
if(nt[i]!=0) {
ans+=nt[i];
if(i+nt[i]>=len)ans--;
}
}
printf("%lld\n",ans);
}
return 0;
}
/*
3
_Happy_New_Year_
ywwyww
zjczzzjczjczzzjc
17
7
32
*/