[后缀数组] BZOJ 2119 股市的预测

类似 [Noi2016]优秀的拆分

枚举 A 的长度 l，把字符串中所有下标为 l 倍数的位置标记出来。左端点 A 的部分一定恰好包含一个标记点。
枚举每一个标记点 i，它的对应点是 i+l+m，后缀数组求出往前
最多对应相同的位数 L 和往后最多位数 R。
贡献为 max(0,min(L-1,l-1)+min(R-1,l-1)+1-l+1);

#include<cstdio>  
#include<cstdlib>  
#include<algorithm>  
#include<cstring>  
#define cl(x) memset(x,0,sizeof(x))  
using namespace std;  
typedef long long ll;  

inline char nc(){  
  static char buf[100000],*p1=buf,*p2=buf;  
  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }  
  return *p1++;   
}
inline void read(int &x){  
  char c=nc(),b=1;  
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;  
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;  
}

const int N=50005;  

struct SArray{  
  int n;    
  int sa[N],t1[N],t2[N],c[N];    
  int height[N],rank[N];    
  inline void SA(int *r,int m){    
    int *x=t1,*y=t2;    
    for (int i=0;i<=m;i++) c[i]=0;    
    for (int i=1;i<=n;i++) c[x[i]=r[i]]++;    
    for (int i=1;i<=m;i++) c[i]+=c[i-1];    
    for (int i=n;i;i--) sa[c[x[i]]--]=i;    
    for (int k=1;k<=n;k<<=1){    
      int p=0;    
      for (int i=n-k+1;i<=n;i++) y[++p]=i;    
      for (int i=1;i<=n;i++) if (sa[i]>k) y[++p]=sa[i]-k;    
      for (int i=0;i<=m;i++) c[i]=0;    
      for (int i=1;i<=n;i++) c[x[y[i]]]++;    
      for (int i=1;i<=m;i++) c[i]+=c[i-1];    
      for (int i=n;i;i--) sa[c[x[y[i]]]--]=y[i];    
      swap(x,y);    
      x[sa[1]]=1; p=1;    
      for (int i=2;i<=n;i++)    
    x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k])?p:++p;    
      if (p>=n) break;    
      m=p;    
    }    
  }    
  inline void GetHeight(int *S,int *sa,int n){    
    int j,k=0;    
    for (int i=1;i<=n;i++) rank[sa[i]]=i;    
    for (int i=1;i<=n;height[rank[i++]]=k)    
      for (k?k--:0,j=sa[rank[i]-1];S[i+k]==S[j+k];)    
    k++;    
  }  
  int st[N][20],Log[N];    
  inline void Pre(int n,int *a){    
    for (int i=2;i<=n;i++) Log[i]=Log[i>>1]+1;    
    for (int i=1;i<=n;i++)    
      st[i][0]=a[i];    
    for (int k=1;k<20;k++)    
      for (int i=1;i<=n;i++){    
    st[i][k]=st[i][k-1];    
    if (i+(1<<(k-1))<=n)    
      st[i][k]=min(st[i][k],st[i+(1<<(k-1))][k-1]);    
      }    
  }    
  inline int Query(int l,int r){  
    if (l>r) swap(l,r);  
    int t=Log[r-l+1];    
    return min(st[l][t],st[r-(1<<t)+1][t]);    
  }  
  inline void Build(int *S,int len){  
    cl(sa); cl(height); cl(rank); cl(t1); cl(t2);  
    n=len; SA(S,n);  
    GetHeight(S,sa,n);  
    Pre(len,height);  
  }  
  inline int LCP(int x,int y){  
    if (x>n || x<=0 || y>n || y<=0) return 0;  
    if (rank[x]>rank[y]) swap(x,y);  
    return Query(rank[x]+1,rank[y]);  
  }  
}suf,pre;

int n,m,a[N];
int sx[N],icnt;
inline int Bin(int x){
  return lower_bound(sx+1,sx+icnt+1,x)-sx;
}

int main(){
  freopen("t.in","r",stdin);
  freopen("t.out","w",stdout);
  read(n); read(m);
  for (int i=1;i<=n;i++) read(a[i]);
  for (int i=1;i<n;i++) a[i]=a[i+1]-a[i],sx[++icnt]=a[i]; n--;
  sort(sx+1,sx+icnt+1); icnt=unique(sx+1,sx+icnt+1)-sx-1;
  for (int i=1;i<=n;i++) a[i]=Bin(a[i]); a[n+1]=0;
  pre.Build(a,n); reverse(a+1,a+n+1); suf.Build(a,n); reverse(a+1,a+n+1);
  ll ans=0;
  for (int l=1;l+l+m<=n;l++)
    for (int i=l;i+l+m<=n;i+=l){
      int R=pre.LCP(i,i+l+m),L=suf.LCP(n-i+1,n-(i+l+m)+1);
      ans+=max(0,min(L-1,l-1)+min(R-1,l-1)+1-l+1);
    }
  printf("%lld\n",ans);
  return 0;
}