bzoj 2119 后缀数组

题意：给一个串，求形如ABA，其中B的长度为m，A的长度不限的子串的数量。

求正串反串的后缀数组和对于h的RMQ。枚举A串长度len，对于每个长度为len的区间处理在这段区间开始的子串个数。

每隔len枚举i，然后用RMQ求i和i+len+m的最长公共前缀后缀l,r。统计长度为len的在区间 [ i-l+1 , i+r-1 ] 内部的长度为len的串个数。

为了避免算重，l和r对len取min。

#include <bits/stdc++.h>
using namespace std;
#define N 101000
#define ll long long
int n,m,cnt;
ll ans;
int a[N],b[N],bir[N];
struct suffix_array
{
    int a[N],sa[N],h[N][21],rank[N],tr[N],has[N];
    int cmp(int x,int y,int k)
    {
        if(x+k>n||y+k>n)return 0;
        return rank[x]==rank[y]&&rank[x+k]==rank[y+k];
    }
    void getsa()
    {
        int i,cnt;
        for(i=1;i<=n;i++)has[a[i]]++;
        for(i=1,cnt=0;i<=n;i++)if(has[i])tr[i]=++cnt;
        for(i=1;i<=n;i++)has[i]+=has[i-1];
        for(i=1;i<=n;i++)rank[i]=tr[a[i]],sa[has[a[i]]--]=i;
        for(int k=1;cnt!=n;k<<=1)
        {
            for(i=1;i<=n;i++)has[i]=0;
            for(i=1;i<=n;i++)has[rank[i]]++;
            for(i=1;i<=n;i++)has[i]+=has[i-1];
            for(i=n;i>=1;i--)if(sa[i]>k)tr[sa[i]-k]=has[rank[sa[i]-k]]--;
            for(i=1;i<=k;i++)tr[n-i+1]=has[rank[n-i+1]]--;
            for(i=1;i<=n;i++)sa[tr[i]]=i;
            for(i=1,cnt=0;i<=n;i++)tr[sa[i]]=cmp(sa[i],sa[i-1],k) ? cnt:++cnt;
            for(i=1;i<=n;i++)rank[i]=tr[i];
        }
        for(int i=1;i<=n;i++)
        {
            if(rank[i]==1)continue;
            for(int j=max(h[rank[i-1]][0]-1,1);;j++)
            {
                if(a[i+j-1]==a[sa[rank[i]-1]+j-1])h[rank[i]][0]=j;
                else break;
            }
        }
        for(int i=1;i<=20;i++)
            for(int j=1;j<=n;j++)
                if(j-(1<<i-1)>0)
                h[j][i]=min(h[j][i-1],h[j-(1<<i-1)][i-1]);
    }   
    int query(int x,int y)
    {
        x=rank[x],y=rank[y];
        if(x>y)swap(x,y);
        int t=bir[y-x];
        return min(h[y][t],h[x+(1<<t)][t]);
    }
}s1,s2;
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    n--;
    for(int i=1;i<=n;i++)b[i]=a[i]=a[i+1]-a[i];
    sort(b+1,b+1+n);
    cnt=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=n;i++)
    {
        a[i]=lower_bound(b+1,b+1+cnt,a[i])-b;
        s1.a[i]=a[i];
        s2.a[n-i+1]=a[i];
    }
    s1.getsa();
    s2.getsa();
    for(int i=1,j=0;i<=n;i++)
    {
        if(1<<(j+1)<=i)j++;
        bir[i]=j;
    }
    for(int len=1;len*2+m<=n;len++)
        for(int i=1;i+len+m<=n;i+=len)
        {
            int r=s1.query(i,i+len+m);
            int l=s2.query(n-i+1,n-(i+len+m)+1);
            r=min(r,len);l=min(l,len);
            ans+=max(r+l-len,0);
        }
    printf("%lld\n",ans);
    return 0;
}