【Caioj1462】字符串的相似度

Source and Judge

Caioj1462

Problem

【Description】
我们定义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,L为字符串S的长度,且S由a-z的小写字母组成。
【Output】
输出S同所有后缀的相似度的和。
【Limited conditions】
1 <= L <= 1000000
【Sample input】
ababaa
【Sample output】
11
【Sample explanation】

Record

20min

Analysis

请先思考

exkmp裸题
ex数组之和
一开始忘开long long了,wa了一个

Code

请先思考
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
//Zory in 2018
//*******************头文件*******************
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
typedef long long ll;
typedef unsigned long long ull;
int mymax(int x,int y) {return x>y?x:y;}
int mymin(int x,int y) {return x<y?x:y;}
int myabs(int x) {return x>0?x:-x;}
void chmax(int &x,int y) {if(x<y) x=y;}
void chmin(int &x,int y) {if(x>y) x=y;}
//*******************全局常量*******************
const int MAXN=1100000;
//*******************全局定义*******************
char sa[MAXN],sb[MAXN];
int la,lb;
int ext[MAXN],ex[MAXN];
//*******************实现*******************
void pre_exkmp()
{
ext[1]=lb;
ext[2]=0;while(2+ext[2]<=lb and sb[1+ext[2]]==sb[2+ext[2]]) ext[2]++;
int k=2,rx=k+ext[k]-1;
for(int i=3;i<=lb;i++)
{
int j=i-k+1,L=ext[j];//对应点
if(L<rx-i+1) ext[i]=L;
else {ext[i]=mymax(rx-i+1,0);while(i+ext[i]<=lb and sb[1+ext[i]]==sb[i+ext[i]]) ext[i]++;}
if(i+ext[i]-1>rx) k=i,rx=k+ext[k]-1;
}
}
void exkmp()
{
pre_exkmp();
ex[1]=0;while(1+ex[1]<=lb and sb[1+ex[1]]==sa[1+ex[1]]) ex[1]++;
int k=1,rx=k+ex[k]-1;
for(int i=2;i<=la;i++)
{
int j=i-k+1,L=ext[j];//对应点
if(L<rx-i+1) ex[i]=L;
else {ex[i]=mymax(rx-i+1,0);while(1+ex[i]<=lb and i+ex[i]<=la and sb[1+ex[i]]==sa[i+ex[i]]) ex[i]++;}
if(i+ex[i]-1>rx) k=i,rx=k+ex[k]-1;
}
}
//*******************主函数*******************
int main()
{
scanf("%s",sa+1);
memcpy(sb,sa,sizeof(sa));
la=lb=strlen(sa+1);
exkmp();
ll ans=0;
for(int i=1;i<=la;i++) ans+=ex[i];
printf("%lld",ans);
}

本文基于 知识共享署名-相同方式共享 4.0 国际许可协议发布
本文地址:http://zory.cf/2018-04/字符串的相似度.html
转载请注明出处,谢谢!

哪怕是一杯奶茶,也将鼓励我继续创作!
0%