【Caioj1178】最长共同前缀长度

Source and Judge

Caioj1178

Problem

【Description】
给出模板串A和子串B,长度分别为lenA和lenB,要求对于每个A[i],(1<=i<=lenA),求出A[i..lenA]与B的最长公共前缀长度
【Input】
输入A,B两个串
【Output】
输出lenA个数,表示A[i…lenA]与B的最长公共前缀长度,每个数之前有空格
【Limited conditions】
lenB<=lenA<=1000000
【Sample input】
aabbabaaab
aabb
【Sample output】
4 1 0 0 1 0 2 3 1 0
【Sample explanation】

Record

2h

Analysis

请先思考

exkmp入门题,求ex数组
详见:【OI之路】10字符串-3ExKMP

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
//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()
{
lb=strlen(sb+1);
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()
{
la=strlen(sa+1);
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()
{
//freopen("tmp.in","r",stdin);
scanf("%s%s",sa+1,sb+1);
pre_exkmp();
exkmp();
for(int i=1;i<=la;i++) printf("%d ",ex[i]);
}

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

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