HDU - 6629 string matching(扩展KMP)

题目链接：点击查看

题目大意：给出一个字符串 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;
}