zoukankan      html  css  js  c++  java
  • bzoj3230

    题解:

    后缀数组

    用height数组求出前i个后缀本质不同的子串个数

    在这个数组中二分可以得到每次询问的俩个子串在原串中的位置

    然后就是求这俩子串的最长公共前缀/最长公共后缀

    st表解决(为了这一题我还看了一下st表)

    ps:注意long long

    代码:

    #include<bits/stdc++.h>
    typedef long long ll;
    using namespace std;
    const int N=100010;
    int n,m,Log[N];
    char ch[N];
    struct SA
    {
        int p,q,k,sa[2][N],rk[2][N],mn[17][N],a[N],h[N],v[N];ll s[N];
        SA(){p=0,q=1;}
        void mul(int *sa,int *rk,int *SA,int *RK)
         {
            for (int i=1;i<=n;i++)v[rk[sa[i]]]=i;
            for (int i=n;i;i--)
             if (sa[i]>k)SA[v[rk[sa[i]-k]]--]=sa[i]-k;
            for (int i=n-k+1;i<=n;i++)SA[v[rk[i]]--]=i;
            for (int i=1;i<=n;i++)
             RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i-1]]!=rk[SA[i]]||rk[SA[i-1]+k]!=rk[SA[i]+k]);
         }
        void getsa()
         {
            for (int i=1;i<=n;i++)v[a[i]]++;
            for (int i=1;i<=30;i++)v[i]+=v[i-1];
            for (int i=1;i<=n;i++)sa[p][v[a[i]]--]=i;
            for (int i=1;i<=n;i++)
             rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i]]!=a[sa[p][i-1]]);
            for (k=1;k<n;k<<=1,swap(p,q))mul(sa[p],rk[p],sa[q],rk[q]);
            for (int k=0,i=1;i<=n;i++)
             {
                int j=sa[p][rk[p][i]-1];
                while (a[i+k]==a[j+k])k++;
                h[rk[p][i]]=k;if(k)k--;            
             }
         }
        void pre()
         {
            for (int i=1;i<=n;i++)a[i]=ch[i]-'a'+1;
            getsa();
            for (int i=1;i<=n;i++)mn[0][i]=h[i];
            for (int i=1;i<=Log[n];i++)
             for (int j=1;j<=n;j++)
              if (j+(1<<i)-1<=n)mn[i][j]=min(mn[i-1][j],mn[i-1][j+(1<<(i-1))]);
              else break;
            for (int i=1;i<=n;i++)s[i]=s[i-1]+n-sa[p][i]+1-h[i];
         }
        int query(int a,int b)
         {
            a=rk[p][a],b=rk[p][b];
            if (a>b)swap(a,b);
            a++;
            int t=Log[b-a+1];
            return min(mn[t][a],mn[t][b-(1<<t)+1]);
         }
        void print()
         {
            for (int i=2;i<=n;i++)printf("%d ",h[i]);
         }
    }A,B;
    int main()
    {
        Log[0]=-1;
        for(int i=1;i<N;i++)Log[i]=Log[i>>1]+1;
        scanf("%lld%lld",&n,&m);
        scanf("%s",ch+1);
        A.pre();
        reverse(ch+1,ch+n+1);
        B.pre();
        for (int i=1;i<=m;i++)
         {
            ll l,r,id,a1,a2,b1,b2,ans=0;
            scanf("%lld%lld",&l,&r);
            if (l>A.s[n]||r>A.s[n])
             {
                 puts("-1");
                continue;
             }
            id=lower_bound(A.s+1,A.s+n+1,l)-A.s;
            a1=A.sa[A.p][id];
            b1=A.sa[A.p][id]+A.h[id]-1+l-A.s[id-1];
            id=lower_bound(A.s+1,A.s+n+1,r)-A.s;
            a2=A.sa[A.p][id];
            b2=A.sa[A.p][id]+A.h[id]-1+r-A.s[id-1];
            ll t=(a1==a2)?2e9:A.query(a1,a2);
            t=min(t,min(b1-a1+1,b2-a2+1));ans+=t*t;
            t=(n-b1+1==n-b2+1)?2e9:B.query(n-b1+1,n-b2+1);
            t=min(t,min(b1-a1+1,b2-a2+1));ans+=t*t;
            printf("%lld
    ",ans);
         }
        return 0;
    }
  • 相关阅读:
    Thinking Clearly about Performance
    国家统计局2018-10-30统计的最新的省市区
    Java 返回字符串中第一个不重复字符的下标 下标从0开始
    国际象棋的游戏规则
    Python语言之requests库
    VBS 自动发消息给对方
    Windows
    上海合计共有16个区一个县
    @RequestMapping中的注解
    Linux 中 /proc/kcore为啥如此之大
  • 原文地址:https://www.cnblogs.com/xuanyiming/p/8490554.html
Copyright © 2011-2022 走看看