文件修复

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]]);
}