-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSAM.cpp
More file actions
82 lines (70 loc) · 1.88 KB
/
SAM.cpp
File metadata and controls
82 lines (70 loc) · 1.88 KB
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#include<bits/stdc++.h>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define Forpiter(x) for(int &p=iter[x];p;p=next[p])
#define Lson (x<<1)
#define Rson ((x<<1)+1)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,127,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define INF (2139062143)
#define F (100000007)
#define MAXN (90000*2+10)
#define Sigmasize (26)
typedef unsigned long long ll;
ll mul(ll a,ll b){return (a*b)%F;}
ll add(ll a,ll b){return (a+b)%F;}
ll sub(ll a,ll b){return (a-b+llabs(a-b)/F*F+F)%F;}
void upd(ll &a,ll b){a=(a%F+b%F)%F;}
class SAM {
public:
char s[MAXN];
int n;
SAM():n(0){MEM(s) MEM(son) MEM(pre) MEM(step) last=tot=0;}
SAM(char *_s){n=strlen(_s),memcpy(s,_s,sizeof(char)*(n+5)); MEM(son) MEM(pre) MEM(step) last=tot=0;}
void mem(){n=0; MEM(s) MEM(son) MEM(pre) MEM(step) last=tot=0;}
void mem(char *_s){n=strlen(_s);memcpy(s,_s,sizeof(char)*(n+5)); MEM(son) MEM(pre) MEM(step) last=tot=0; }
int son[MAXN][Sigmasize+1],pre[MAXN],step[MAXN],last,tot;
void extend(char ch)
{
step[++tot]=step[last]+1;
int p=last,np=tot;
for(;!son[p][ch];p=pre[p]) son[p][ch]=np;
if (!p) pre[np]=1;
else {
int q=son[p][ch];
if (step[q]==step[p]+1) pre[np]=q;
else {
step[++tot]=step[p]+1;
int nq=tot;
memcpy(son[nq],son[q],sizeof(son[q]));
pre[nq]=pre[q];
pre[q]=pre[np]=nq;
for(;son[p][ch]==q;p=pre[p]) son[p][ch]=nq;
}
}
last=np;
}
void build(){
last=tot=1;
Rep(i,n) extend(s[i]-'a');
}
}S1,S2;
char s1[MAXN],s2[MAXN];
int main()
{
// freopen(".in","r",stdin);
int T;cin>>T;
while(T--) {
scanf("%s%s",s1,s2);
S1.mem(s1),S2.mem(s2);
S1.build();
S2.build();
}
return 0;
}