bzoj4556: [Tjoi2016&Heoi2016]字符串 (后缀数组加主席树)

题目是给出一个字符串，每次询问一个区间[a,b]中所有的子串和另一个区间[c,d]的lcp最大值，首先求出后缀数组，对于lcp的最大值肯定是rank[c]的前驱和后继，但是对于这个题会出现问题，就是题目中有区间的限制。

For example:

5 1

aaaab

1 2 3 5

对于这个样例，如果直接找到aab的前驱是 aaab，然后由于区间的原因答案是1，但是如果我们再往前找的话，找到aaaab，答案会变成2。那就出现了错误。考虑一下怎么做可以去除这种影响呢？

我们可以二分一下，首先对于[a,b]这个区间我们只考虑前一半区间[a,mid] 我们利用上面求前驱和后继的方法求出一个答案，如果答案大于mid我们就要考虑mid以前的区间，反之就考虑mid以后的区间。(这里的区间是指子串的左端点所在的区间)

为什么是对的？分类讨论一下，如果答案大于mid，那么我们不可能会考虑mid之后的区间，因为显然mid之后的区间到b的长度都小于mid，答案不会变优。如果答案小于mid，我们就不会在考虑mid之前的区间，因为算出答案小于mid，显然它不可能是由于区间限制得出的，因为我们只考虑前一半的区间，它到末尾的长度肯定大于等于mid，所以我们往后二分，因为随着区间的扩大我们找到的前驱和后继肯定会越来越接近rank[c]，那么答案可能会变优，而我们往前判断答案肯定不会变优，所以往后二分。这样二分的正确性就ok了。（其实这个性质挺不好想的，我也是懵逼了半天，然而大奕哥用一个暴力碾掉了标算，bzoj上rank1......详情可以看博客 http://www.cnblogs.com/nbwzyzngyl/p/8215412.html  纯属侥幸......）

那么我们可以用主席树来办到求前驱和后继，这道题就解决了。—— by VANE

#include<bits/stdc++.h>
using namespace std;
const int N=;
const int M=;
int n,m,a[N],tax[N],tp[N],height[N],st[N][],Log[N];
int sa[N],rank[N],sum[M],ls[M],rs[M],siz,root[N],mm;
char s[N];
void rsort()
{
    for(int i=;i<=m;++i) tax[i]=;
    for(int i=;i<=n;++i) tax[rank[tp[i]]]++;
    for(int i=;i<=m;++i) tax[i]+=tax[i-];
    for(int i=n;i;--i) sa[tax[rank[tp[i]]]--]=tp[i];
}
bool cmp(int *f,int x,int y,int w){return max(x,y)+w<=n&&f[x]==f[y]&&f[x+w]==f[y+w];}
void suffix()
{
    m=;for(int i=;i<=n;++i) tp[i]=i;
    for(int i=;i<=n;++i) rank[i]=a[i];
    rsort();
    for(int w=,p;w<n;w+=w)
    {
        p=;for(int i=n-w+;i<=n;++i) tp[++p]=i;
        for(int i=;i<=n;++i) if(sa[i]>w) tp[++p]=sa[i]-w;
        rsort();swap(rank,tp);rank[sa[]]=p=;
        for(int i=;i<=n;++i) rank[sa[i]]=cmp(tp,sa[i],sa[i-],w)?p:++p;
        m=p;if(p>=n) break;
    }
    int j,k=;
    for(int i=;i<=n;height[rank[i++]]=k)
        for(k=k?k-:,j=sa[rank[i]-];a[j+k]==a[i+k];++k);
    memset(st,,sizeof st);
    for(int i=;i<=n;++i) st[i][]=height[i];
    for(int j=;j<=;++j)
        for(int i=;i+(<<j)-<=n;++i)
            st[i][j]=min(st[i][j-],st[i+(<<j-)][j-]);
    for(int i=,j=;i<=n;++i)
    {
        if(i==(<<j+)) ++j;
        Log[i]=j;
    }
}
int lcp(int x,int y)
{
    if(x>y) swap(x,y);
    int len=Log[y-x+];
    int ret=min(st[x][len],st[y-(<<len)+][len]);
    return ret;
}
void insert(int l,int r,int x,int &y,int z)
{
    y=++siz;
    sum[y]=sum[x]+;
    ls[y]=ls[x];rs[y]=rs[x];
    if(l==r) return;
    int mid=l+r>>;
    if(z<=mid) insert(l,mid,ls[x],ls[y],z);
    else insert(mid+,r,rs[x],rs[y],z);
}
int query_sum(int l,int r,int x,int y,int le,int ri)
{
    if(le>ri) return ;
    if(l==le&&r==ri) return sum[y]-sum[x];
    int mid=l+r>>;
    if(ri<=mid) return query_sum(l,mid,ls[x],ls[y],le,ri);
    if(le>mid) return query_sum(mid+,r,rs[x],rs[y],le,ri);
    return query_sum(l,mid,ls[x],ls[y],le,mid)+query_sum(mid+,r,rs[x],rs[y],mid+,ri);
}
int query_pos(int l,int r,int x,int y,int p)
{
    if(l==r) return l;
    int mid=l+r>>;
    if(sum[ls[y]]-sum[ls[x]]>=p) return query_pos(l,mid,ls[x],ls[y],p);
    return query_pos(mid+,r,rs[x],rs[y],p-sum[ls[y]]+sum[ls[x]]);
}
int main()
{
    //freopen("str.in","r",stdin);
    //freopen("str.out","w",stdout);
    scanf("%d%d",&n,&mm);
    scanf("%s",s+);for(int i=;i<=n;++i) a[i]=s[i];
    suffix();
    int a,b,c,d,x,pre,sub;
    for(int i=;i<=n;++i) insert(,n,root[i-],root[i],rank[i]);
    while(mm--)
    {
        scanf("%d%d%d%d",&a,&b,&c,&d);
        int l=,r=b-a+,maxn=,mid;
        bool flag=;
        while(l<=r)
        {
            mid=l+r>>;
            x=query_sum(,n,root[a-],root[b-mid],,rank[c]-);
            pre=x?query_pos(,n,root[a-],root[b-mid],x):-;
            x=query_sum(,n,root[a-],root[b-mid],,rank[c]);
            sub=query_pos(,n,root[a-],root[b-mid],+x);
            if(x==sum[root[b-mid]]-sum[root[a-]]) sub=-;
            int ans=;
            if(pre!=-)
            {
                int o=min(lcp(pre+,rank[c]),b-sa[pre]+);
                ans=max(ans,min(o,d-c+));
            }
            if(sub!=-)
            {
                int o=min(lcp(rank[c]+,sub),b-sa[sub]+);
                ans=max(ans,min(o,d-c+));
            }
            if(c>=a&&c<=b) ans=max(ans,min(d,b)-c+);
            maxn=max(maxn,ans);
            if(ans>=mid) l=mid+;
            else r=mid;
            if(flag) break;
            if(l==r) flag=;
        }
        printf("%d\n",maxn);
    }
}