Description
给出一个字符串S,求S有多少个出现了2次以上的字串。
|S|<=100000
Solutioin
又是裸题,求出SA用单调栈直接搞就好了。
用来复习SA
Code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define N 100005
using namespace std;
int n,ans,ws[N],sa[N],rank[N],height[N],x[N],y[N],d[N];
char st[N];
void tsort() {
memset(ws,0,sizeof(ws));int mx=0;
fo(i,1,n) ws[x[y[i]]]++,mx=max(mx,x[y[i]]);
fo(i,1,mx) ws[i]+=ws[i-1];
fd(i,n,1) sa[ws[x[y[i]]]]=y[i],ws[x[y[i]]]--;
}
void getsa() {
fo(i,1,n) y[i]=i,ws[x[i]]++;tsort();
for(int j=1;j<=n;j*=2) {
int k=0;fo(i,n-j+1,n) y[++k]=i;
fo(i,1,n) if (sa[i]>j) y[++k]=sa[i]-j;
tsort();
fo(i,1,n) y[i]=x[i],x[i]=0;
x[sa[1]]=1;k=1;
fo(i,2,n) {
if (y[sa[i-1]]!=y[sa[i]]||y[sa[i-1]+j]!=y[sa[i]+j]) k++;
x[sa[i]]=k;
}
if (k==n) break;
}
fo(i,1,n) rank[sa[i]]=i;
}
void getheight() {
int k=0;
fo(i,1,n) {
if (k) k--;
int j=sa[rank[i]-1];
while (i+k<=n&&j+k<=n&&st[i+k]==st[j+k]) k++;
height[rank[i]]=k;
}
}
int main() {
scanf("%s",st+1);n=strlen(st+1);
fo(i,1,n) x[i]=st[i];
getsa();getheight();
fo(i,1,n) {
int x=height[i];
while (d[0]&&d[d[0]]>=height[i]) x=max(x,d[d[0]--]);
ans+=x-height[i];d[++d[0]]=height[i];
}
printf("%d",ans+d[d[0]]);
}