BZOJ 4556 [HEOI 2016] str

来源：互联网发布：win7恢复网络设置方法编辑：程序博客网时间：2024/06/06 18:49

学习了一下经典的在后缀数组上维护主席树的思想。
求出sa和height数组，以sa数组对原串建主席树，对每个询问二分答案出长度ans，然后只要在（c,c+len）的区间中查找是否有属于(a,b-len+1)的下标。

出了几个小问题查了好久。

开始用的cmath 里的log函数求RMQ 直接爆炸，还是自己手打吧。

小样例全过了但交上去WA了，发现自己RMQ的st数组…st[maxn][20]…
maxn和20居然写反了。。。我说咋老RE…

#include<cstdio>#include<iostream>#include<cstring>#include<cmath>using namespace std;const int maxn=103010;int c[maxn],t1[maxn],t2[maxn],sa[maxn],height[maxn],root[maxn],rank[maxn],cnt,n,m,st[20][maxn],lg[maxn];char s[maxn];void build_sa(int m){    int *x=t1,*y=t2;    for(int i=0;i<m;i++) c[i]=0;    for(int i=0;i<n;i++) c[x[i]=s[i]]++;     for(int i=0;i<m;i++) c[i]+=c[i-1];    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;    for(int k=1;k<=n;k<<=1)    {        int p=0;        for(int i=n-k;i<n;i++) y[p++]=i;        for(int i=0;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=0;i<n;i++) c[x[i]]++;        for(int i=0;i<m;i++) c[i]+=c[i-1];        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];        swap(x,y);        p=1,x[sa[0]]=0;        for(int i=1;i<n;i++) x[sa[i]]= y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k] ? p-1 : p++;        if(p>=n) break;        m=p;     }   }void get_height(){    int j,k=0;    for(int i=1;i<=n;i++) rank[sa[i]]=i;    for(int i=0;i<n;i++)    {        if(k) k--;        j=sa[rank[i]-1];        while(s[i+k]==s[j+k]) k++;        height[rank[i]]=k;    }}struct node{    int l,r,sum;};node T[maxn*40];void update(int l,int r,int &x,int &y,int pos){    x=++cnt;T[x]=T[y];T[x].sum++;     int mid=(l+r)>>1;    if(l==r) return ;    if(pos<=mid) update(l,mid,T[x].l,T[y].l,pos);    else update(mid+1,r,T[x].r,T[y].r,pos);} int query(int l,int r,int x,int y,int L,int R){    if(L<=l&&r<=R) return T[y].sum-T[x].sum;    int ans=0;    int mid=(l+r)>>1;    if(L<=mid) ans=query(l,mid,T[x].l,T[y].l,L,R);    if(R>mid) ans+=query(mid+1,r,T[x].r,T[y].r,L,R);    return ans;}void St() {    for(int i=1;i<=n;i++) st[0][i] = height[i];    for(int j=1;j<18;j++)     for(int i = n-(1<<j)+1; i > 0; -- i)        st[j][i] = min(st[j-1][i], st[j-1][i+(1<<j-1)]);}int a1,b1,c1,d1;int main(){    freopen("heoi2016_str.in","r",stdin);    freopen("heoi2016_str.out","w",stdout);    scanf("%d%d",&n,&m);    scanf("%s",s);n++;    build_sa(200);n--;    get_height();St();    for(int i=1;i<=n;i++) update(1,n,root[i],root[i-1],sa[i]+1);    for(int i=2;i<=n;i++) lg[i] = lg[i>>1] + 1;    for(int i = 1; i <= m; ++ i)     {        scanf("%d%d%d%d",&a1,&b1,&c1,&d1);        int l=1,r=min(d1-c1+1, b1-a1+1), mid, ans = 0;        while(l <= r)         {            int mid=l+r>>1;            int lp = rank[c1-1], rp = lp;            for(int k=lg[n];k>=0;k--)             {                if(lp>=(1<<k)&&st[k][lp-(1<<k)+1]>=mid) lp -= 1<<k;                if(rp<=n-(1<<k)&&st[k][rp+1] >= mid) rp += 1<<k;            }            if(query(1,n,root[lp-1],root[rp],a1,b1-mid+1)>0) ans = mid, l = mid+1;            else r = mid-1;        }        printf("%d\n", ans);    }    return 0;}

0 0