bzoj2119 股市的预测(SA+lcp)

走势相同，要进行差分，数的范围太大，还要离散化，然后就是求有多少子串样子是ABA，且B的长度已给定。枚举A的长度x，统计所有A的长度为x的答案。
我们以x对原串进行分块，每一块的第一个点设为关键点，我们发现对于一个ABA，第一个A一定覆盖且只覆盖一个关键点，我们枚举关键点i，i和i+mm+x往左往右匹配长度l和r(r包括关键点)，如果l+r>=L那么这个地方就有l+r-x+1个满足的子串。因为要避免重复，也就是要保证 第一个A覆盖且只覆盖一个关键点，所以我们的r最多匹配x，l最多匹配x-1.复杂度$O(nlogn)$，分析同poj3693

#include <bits/stdc++.h>
using namespace std;
#define N 50010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,a[N],aa[N],mm,Log[N],ans=0;
struct SA{
    int s[N],rank[N<<1],rank1[N],sa[N],h[N],tmp[N],count[N],st[N][20];
    void init(){
        for(int i=1;i<=m;++i) count[i]=0;
        for(int i=1;i<=n;++i) count[s[i]]=1;
        for(int i=1;i<=m;++i) count[i]+=count[i-1];
        for(int i=n;i>=1;--i) rank[i]=count[s[i]];
        int k=0;
        for(int p=1;k!=n;p<<=1,m=k){
            for(int i=1;i<=m;++i) count[i]=0;
            for(int i=1;i<=n;++i) count[rank[i+p]]++;
            for(int i=1;i<=m;++i) count[i]+=count[i-1];
            for(int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;
            for(int i=1;i<=m;++i) count[i]=0;
            for(int i=1;i<=n;++i) count[rank[tmp[i]]]++;
            for(int i=1;i<=m;++i) count[i]+=count[i-1];
            for(int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];
            memcpy(rank1,rank,sizeof(rank1));
            rank[sa[1]]=k=1;
            for(int i=2;i<=n;++i){
                if(rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;
                rank[sa[i]]=k;
            }
        }k=0;
        for(int i=1;i<=n;++i){
            if(rank[i]==1){h[1]=0;continue;}
            if(i==1||h[rank[i-1]]<=1) k=0;
            if(k) --k;
            while(s[i+k]==s[sa[rank[i]-1]+k]) ++k;
            h[rank[i]]=k;
        }
        for(int i=1;i<=n;++i) st[i][0]=h[i];
        for(int i=1;i<=Log[n];++i)
            for(int j=1;j<=n;++j)
                if(j+(1<<i-1)<=n) st[j][i]=min(st[j][i-1],st[j+(1<<i-1)][i-1]);
    }
    int lcp(int x,int y){
        int l=rank[x],r=rank[y];
        if(l>r) swap(l,r);l++;
        int t=Log[r-l+1];
        return min(st[l][t],st[r-(1<<t)+1][t]);
    }
}A,B;
inline void solve(int x){
    for(int i=1;i+x+mm<=n;i+=x){
        int j=i+x+mm;
        int r=min(A.lcp(i,j),x),l=min(B.lcp(n+1-i+1,n+1-j+1),x-1);
        if(l+r>=x) ans+=l+r-x+1;
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();mm=read();Log[0]=-1;
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;i<=n;++i) a[i]=read();
    for(int i=1;i<n;++i) aa[i]=a[i]=a[i+1]-a[i];n--;
    sort(aa+1,aa+n+1);
    m=unique(aa+1,aa+n+1)-aa-1;
    for(int i=1;i<=n;++i) a[i]=lower_bound(aa+1,aa+m+1,a[i])-aa;
    for(int i=1;i<=n;++i) A.s[i]=a[i],B.s[n-i+1]=a[i];
    A.init();B.init();
    for(int i=1;i+i+mm<=n;++i) 
    solve(i);
    printf("%d\n",ans);
    return 0;
}