首页 > 代码库 > BZOJ 2119: 股市的预测
BZOJ 2119: 股市的预测
Description
求形如ABA形式的字符串,其中B长度固定,\(n\leqslant 10^5\)
Solution
后缀数组。
我们可以枚举一个长度\(x\),然后将序列分组,每组长度为\(x\),然后从\(i\)找和\(i+x+B\)的最长公共后缀和最长公共前缀,然后得到一组合法区间,限制一下在块中防止重复即可。
复杂度就是调和级数。
复杂度\(O(nlogn)\)
Code
/************************************************************** Problem: 2119 User: BeiYu Language: C++ Result: Accepted Time:1680 ms Memory:14972 kb****************************************************************/ #include <bits/stdc++.h>using namespace std; const int N = 100050;const int M = 25; inline int in(int x=0,char ch=getchar()) { while(ch>‘9‘||ch<‘0‘) ch=getchar(); while(ch>=‘0‘&&ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();return x; } int l,n,m,B;long long ans;int a[N],b[N]; namespace SA { int t1[N],t2[N],c[N],sa[N],rk[N],ht[N]; int pw2[N],lg2[N]; int st[N][M]; void get_sa(int a[],int n=::n,int m=::m) { int *x=t1,*y=t2; for(int i=1;i<=m;i++) c[i]=0; for(int i=1;i<=n;i++) c[x[i]=a[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,p=0;k<n;k<<=1,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=1;i<=m;i++) c[i]=0; for(int i=1;i<=n;i++) c[x[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]]=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; } } void get_ht(int a[],int n=::n) { for(int i=1;i<=n;i++) rk[sa[i]]=i; for(int i=1,j,k=0;i<=n;ht[rk[i++]]=k) for(j=sa[rk[i]-1],k=k?k-1:k;a[i+k]==a[j+k];k++); } void get_st(int n=::n) { pw2[0]=1;for(int i=1;i<M;i++) pw2[i]=pw2[i-1]<<1; lg2[0]=-1;for(int i=1;i<=n;i++) lg2[i]=lg2[i>>1]+1; for(int i=1;i<=n;i++) st[i][0]=ht[i]; for(int j=1;j<M;j++) for(int i=1;i<=n;i++) if(i+pw2[j]-1<=n) st[i][j]=min(st[i][j-1],st[i+pw2[j-1]][j-1]); } int lcp(int x,int y) { x=rk[x],y=rk[y]; if(x==y) return n-sa[x]+1; if(x>y) swap(x,y);x++; int lg=lg2[y-x+1]; return min(st[x][lg],st[y-pw2[lg]+1][lg]); }}using namespace SA; void Solve(int x) {// cout<<"x:"<<x<<endl; for(int i=1;i+x+B<=n;i+=x) { int l=lcp(n+n+2-i+1,n+n+2-i-x-B+1),r=lcp(i,i+x+B); l=min(l,x-1),r=min(r,x);// cout<<i<<" "<<i+x+B<<"-->"<<n+n+2-i+1<<" "<<n+n+2-i-x-B+1<<endl;// cout<<i<<" "<<i+B+x<<" "<<l<<" "<<r<<endl; if(r+l>=x) ans+=r+l-x+1; }}int main() { l=n=in(),B=in(); for(int i=1;i<=n;i++) b[i]=in(); for(int i=1;i<n;i++) a[i]=b[i+1]-b[i]; for(int i=1;i<n;i++) b[i]=a[i]; sort(a+1,a+n); m=unique(a+1,a+n)-a-1; for(int i=1;i<n;i++) b[i]=lower_bound(a+1,a+m,b[i])-a; b[n]=m+1; for(int i=n+1;i<n+n;i++) b[i]=b[n+n-i]; m++; for(int i=1;i<n+n;i++) a[i]=b[i]; get_sa(a,n+n-1,m); get_ht(a,n+n-1); get_st(n+n-1); // for(int i=1;i<n+n;i++) cout<<a[i]<<" ";cout<<endl; n--; for(int i=1;i+i+B<=n;i++) Solve(i); printf("%lld\n",ans); return 0;}
BZOJ 2119: 股市的预测
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。