[2018 ACM-ICPC 焦作赛区网络赛] H - String and Times（后缀自动机）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MaxN=2e5+100;
const ll MAXN = MaxN;
ll cntA[MaxN],cntB[MaxN],tsa[MAXN],A[MAXN],B[MAXN];
ll sa[MAXN],Rank[MAXN],h[MAXN];
char ch[MAXN];
struct Node{
    ll val,index;
    Node(ll val_,ll index_):val(val_),index(index_){
    }
    bool operator < (const Node b)const{
        if (val==b.val){
            return b.index<index;
        }
        return b.val<val;
    }
};
priority_queue<Node>pq;
void GetSa(char *ch,ll *sa,ll *rank,ll n){
 
    for(ll i=0;i<MaxN;i++)  cntA[i]=0;
    for(ll i=1;i<=n;i++)   cntA[ch[i]]++;
    for(ll i=1;i<=MaxN;i++) cntA[i]+=cntA[i-1];
    for(ll i=n;i;i--)  sa[cntA[ch[i]]--]=i;
    rank[sa[1]]=1;
    for(ll i=2;i<=n;i++){
        rank[sa[i]]=rank[sa[i-1]];
        if(ch[sa[i]]!=ch[sa[i-1]])  rank[sa[i]]++;
    }
    for(ll l=1;rank[sa[n]]<n;l<<=1){
        for(ll i=0;i<MaxN;i++)  cntA[i]=0;
        for(ll i=0;i<MaxN;i++)  cntB[i]=0;
        for(ll i=1;i<=n;i++){
            cntA[A[i]=rank[i]]++;
            cntB[B[i]=(i+l<=n)?rank[i+l]:0]++;
        }
        for(ll i=1;i<MaxN;i++)   cntB[i]+=cntB[i-1];
        for(ll i=n;i;i--)  tsa[cntB[B[i]]--]=i;
        for(ll i=1;i<MaxN;i++)  cntA[i]+=cntA[i-1];
        for(ll i=n;i;i--)  sa[cntA[A[tsa[i]]]--]=tsa[i];
        rank[sa[1]]=1;
        for(ll i=2;i<=n;i++){
            rank[sa[i]]=rank[sa[i-1]];
            if(A[sa[i]]!=A[sa[i-1]] || B[sa[i]]!=B[sa[i-1]])    rank[sa[i]]++;
        }
    }
}
 
void GetHeight(char *ch,ll *sa,ll *rank,ll *height,ll n){
 
    GetSa(ch,sa,rank,n);
    for(ll i=1,j=0;i<=n;i++){
        if(j)   j--;
        while(ch[i+j]==ch[sa[rank[i]-1]+j]) j++;
        height[rank[i]]=j;
    }
}
ll GetK(ll k,ll n){
    ll ans=0;
    k--;
    if(k==0){
        for(ll i=1;i<=n;++i)   ans=ans+(n-sa[i]+1-h[i]);
        return ans;
    }
    while (!pq.empty())pq.pop();
    for (ll i=2;i<=n;i++){
        while (!pq.empty()&&pq.top().index<i-k+1)pq.pop();
        pq.push(Node(h[i],i));
        if (i>k){
            ll top = pq.top().val;
            ll last = h[i-k];
            ans +=max((ll)0,top-last);
        }
    }
    return ans;
}
 
void Run(){
    ll n,k,m;
    while(~scanf("%s %lld %lld",ch+1,&k,&m)){
        n=strlen(ch+1);
        GetHeight(ch,sa,Rank,h,n);
        printf("%lld\n",GetK(k,n)-GetK(k+m+1,n));  //至少出现n次的字串数-至少出现m+1次的字串数
    }
}
int main(){
    Run();
    return 0;
}