51 nod 1304 1304 字符串的相似度(exkmp)

1304 字符串的相似度


题目来源： HackerRank

基准时间限制：1 秒 空间限制：131072 KB 分值: 320 难度：7级算法题


 收藏


 关注

我们定义2个字符串的相似度等于两个串的相同前缀的长度。例如 "abc" 同 "abd" 的相似度为2，"aaa" 同 "aaab" 的相似度为3。
给出一个字符串S，计算S同他所有后缀的相似度之和。例如：S = "ababaa"，所有后缀为：

ababaa 6
babaa 0
abaa 3
baa 0
aa 1
a 1

S同所有后缀的相似度的和 = 6 + 0 + 3 + 0 + 1 + 1 = 11

Input

输入一个字符串S(1 <= L <= 1000000)，L为字符串S的长度，且S由a-z的小写字母组成。

Output

输出S同所有后缀的相似度的和。

Input示例

ababaa

Output示例

11

#include <iostream>
#include <cstdio>
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6+10;
const LL mod = 1e9+7;
char s1
, s2
;
int nex
, nex1
;
void getnext(char a[],int len)
{
int k=0, i=1;
nex[0]=len;
while(k+1<len&&a[k]==a[k+1])++k;
nex[1]=k;
k=1;
while(++i<len)
{
int p=k+nex[k]-1;
nex[i]=min(nex[i-k],max(p-i+1,0));
while(i+nex[i]<len&&a[nex[i]]==a[i+nex[i]]) ++nex[i];
if(i+nex[i]>k+nex[k]) k=i;
}
return ;
}

int main()
{
scanf("%s",s1);
int len=strlen(s1);
getnext(s1,len);
LL ans=0;
for(int i=0;i<len;i++)
{
ans=ans+(LL)nex[i];
}
printf("%lld\n",ans);
return 0;
}