【bzoj 3238】差異

傳送門~

解題思路

求後綴數組以後搞個單調棧。
因爲height數組的意義讓這個單調棧很煩，注意要理清思路再寫。
代碼：

#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstring>
#include<string>
#include<cstdio>
#include<cstdlib>
using namespace std;
char s[500005];
int x[500005],y[500005];
int tmp[500005],sum[500005];
int rank[500005],sa[500005];
int height[500005];
int my_right[500005];
int my_left[500005];
int my_stack[500005];
int len;
void get_rank(){
    for(int i=1;i<=len;i++) sum[s[i]]++;
    for(int i=1;i<=127;i++) sum[i]+=sum[i-1];
    for(int i=len;i>=1;i--) tmp[sum[s[i]]--]=i;
    int now=0;
    for(int i=1;i<=len;i++){
        if(i==1 || s[tmp[i]]!=s[tmp[i-1]]) ++now;
        rank[tmp[i]]=now;
    }
}
void ji_sort(int *key,int *order){
    for(int i=0;i<=len;i++) sum[i]=0;
    for(int i=1;i<=len;i++) sum[key[i]]++;
    for(int i=1;i<=len;i++) sum[i]+=sum[i-1];
    for(int i=len;i>=1;i--) tmp[sum[key[order[i]]]--]=order[i];
    for(int i=1;i<=len;i++) order[i]=tmp[i];
}
void get_hou(){
    get_rank();
    for(int k=1;k<=len;k<<=1){
        for(int i=1;i<=len;i++){
            x[i]=rank[i];
            y[i]=i+k>len ? 0 : rank[i+k];
            sa[i]=i;
        }
        ji_sort(y,sa);
        ji_sort(x,sa);
        int now=0;
        for(int i=1;i<=len;i++){
            if(i==1 || x[sa[i]]!=x[sa[i-1]] || y[sa[i]]!=y[sa[i-1]]) ++now;
            rank[sa[i]]=now;
        }
    }
}
void calheight(){
    for(int i=1;i<=len;i++){
        if(rank[i]==1) continue;
        int j=max(height[rank[i-1]]-1,0),k=sa[rank[i]-1];
        while(s[k+j]==s[i+j]) j++;
        height[rank[i]]=j;
    }
}
int main(){
    scanf("%s",s+1);
    len=strlen(s+1);
    get_hou();calheight();
    int top=0;
    for(int i=1;i<=len;i++){
        while(height[my_stack[top]]>height[i] && top>0) top--;
        if(top==0) my_left[i]=1;
        else my_left[i]=my_stack[top]+1;
        my_stack[++top]=i;
    }
    top=0;
    for(int i=len;i>=1;i--){
        while(height[my_stack[top]]>=height[i] && top>0) top--;
        if(top==0) my_right[i]=len;
        else my_right[i]=my_stack[top]-1;
        my_stack[++top]=i;
    }
    long long o=len;
    long long ans=(o-1)*(o+1)*o/2;
    long long t;
    for(int i=1;i<=len;i++){
        t=2*(i-my_left[i]+1);
        t*=my_right[i]-i+1;
        t*=height[i];
        ans-=t;
    }
    printf("%lld",ans);
    return 0;
}