题目链接:点击查看
题目大意:给出一个字符串 s 和一个暴力程序,用于求解 s 的每个后缀和原字符串的最长公共前缀,现在问一共需要执行多少次比较操作
题目分析:首先肯定不能暴力去模拟,时间复杂度n*n,也不能用后缀数组,这个题好像特地卡了后缀数组的时间,倍增法nlogn会被卡掉,DC3因为常数太大也会被卡掉,网上说AC自动机会被卡内存,因为整个字符集是ASCII码的范围,那就只能用扩展KMP来求解了,扩展KMP在O(n)时间内求解出extend[ i ]数组,表示字符串 s 的每个后缀和字符串 t 匹配的最长公共前缀,这样一来就和这个题目对应上了,将字符串 t 的位置替换上字符串 s 就是答案了,记得特判一下每次匹配是否匹配到末尾,如果不是末尾需要加一,因为最后那次匹配失败也算一次匹配操作
代码:
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<sstream>
using namespace std;
typedef long long LL;
const int inf=0x3f3f3f3f;
const int N=1e6+100;
char s[N];
LL Next[N],extend[N];
//预处理计算Next数组
void getNext(char str[])
{
int i=0,j,po,len=strlen(str);
Next[0]=len; //初始化Next[0]
while(str[i]==str[i+1] && i+1<len) i++; Next[1]=i; //计算Next[1]
po=1; //初始化po的位置
for(i=2;i<len;i++)
{
if(Next[i-po]+i < Next[po]+po) //第一种情况,可以直接得到Next[i]的值
Next[i]=Next[i-po];
else //第二种情况,要继续匹配才能得到Next[i]的值
{
j = Next[po]+po-i;
if(j<0) j=0; //如果i>po+Next[po],则要从头开始匹配
while(i+j<len && str[j]==str[j+i]) j++; Next[i]=j;
po=i; //更新po的位置
}
}
}
//计算extend数组
void EXKMP(char s1[],char s2[])
{
int i=0,j,po,len=strlen(s1),l2=strlen(s2);
getNext(s2); //计算子串的Next数组
while(s1[i]==s2[i] && i<l2 && i<len) i++; extend[0]=i;
po=0; //初始化po的位置
for(i=1;i<len;i++)
{
if(Next[i-po]+i < extend[po]+po) //第一种情况,直接可以得到extend[i]的值
extend[i]=Next[i-po];
else //第二种情况,要继续匹配才能得到extend[i]的值
{
j = extend[po]+po-i;
if(j<0) j=0; //如果i>extend[po]+po则要从头开始匹配
while(i+j<len && j<l2 && s1[j+i]==s2[j]) j++; extend[i]=j;
po=i; //更新po的位置
}
}
}
int main()
{
// freopen("input.txt","r",stdin);
// ios::sync_with_stdio(false);
int w;
cin>>w;
while(w--)
{
scanf("%s",s);
int n=strlen(s);
getNext(s);
EXKMP(s,s);
LL ans=0;
for(int i=1;i<n;i++)
ans+=extend[i]+(extend[i]!=n-i);
printf("%lld\n",ans);
}
return 0;
}
Frozen_Guardian 发布了578 篇原创文章 · 获赞 18 · 访问量 1万+ 私信 关注