2019年华南理工大学程序设计竞赛（春季赛）第K题 Parco_Love_String

题目

题解

后缀数组暴力水过。。。
时间复杂度O（n^3)

AC代码

#include <bits/stdc++.h>
using namespace std;
const int maxn=1100;
int ranka[maxn+1],tmp[maxn+1],n,k;
bool compare_sa(int i,int j)
{
    if(ranka[i]!=ranka[j]) return ranka[i]<ranka[j];
    else
    {
        int ri=i+k<=n?ranka[i+k]:-1;
        int rj=j+k<=n?ranka[j+k]:-1;
        return ri<rj;
    }
}
void construct_sa(string s,int *sa)
{
    n=s.size();
    for(int i=0;i<=n;i++)
    {
        sa[i]=i;
        ranka[i]=i<n?s[i]:-1;
    }
    for(k=1;k<=n;k*=2)
    {
        sort(sa,sa+n+1,compare_sa);
        //tmp不能省
        tmp[sa[0]]=0;
        for(int i=1;i<=n;i++)
        {
            tmp[sa[i]]=tmp[sa[i-1]]+(compare_sa(sa[i-1],sa[i])?1:0);//括号必须加
        }
        for(int i=0;i<=n;i++) ranka[i]=tmp[i];
    }
}
void construct_lcp(string s,int *sa,int *lcp)
{
    int n=s.length();
    for(int i=0;i<=n;i++) ranka[sa[i]]=i;//取消相同排名

    int h=0;
    lcp[0]=0;//空串与第一个后缀的最长公共前缀为空
    for(int i=0;i<n;i++)
    {
        //计算字符串中从位置i开始的后缀和其在后缀数组中的前一个后缀的LCP
        int j=sa[ranka[i]-1];

        //将h先减去首字母的1长度，在保持前缀相同前提下不断增加
        if(h>0) h--;
        for(;j+h<n && i+h<n;h++)
        {
            if(s[j+h]!=s[i+h]) break;
        }

        lcp[ranka[i]-1]=h;
    }
}

int ans[1100];
int main()
{
   // ios::sync_with_stdio(false);
    string s;

    int sa[maxn],lcp[maxn];
    while(cin>>s)
    {
        int len=s.size();
        //求后缀数组
        construct_sa(s,sa);
        //求高度数组
        construct_lcp(s,sa,lcp);
        //输出
        int T;
        //cin>>T;
        scanf("%d",&T);
        memset(ans,0,sizeof(ans));
        while(T--)
        {
            int pos,sum=0;
            //cin>>pos;
            scanf("%d",&pos);
            pos--;
            if(ans[pos]){
                //cout<<ans[pos]<<endl;
                printf("%d\n",ans[pos]);
                continue;
            }
            for(int i=1;i<len;i++)
            {
                int now=lcp[i];
                for(int j=i+1;j<=len;j++)
                {
                    now=min(now,lcp[j-1]);
                    if((sa[i]<=pos)!=sa[j]<=pos)
                    {
                        int x=sa[i]<=pos?i:j;
                        sum+=min(pos-sa[x]+1,now);
                    }
                    if(now==0) break;
                }
            }
            ans[pos]=sum;
            //cout<<sum<<endl;
            printf("%d\n",sum);

        }
       // for(int i=0;i<=len;i++)
       // {
        //    cout<<i<<" "<<sa[i]<<" "<<lcp[i]<<endl;
       // }
    }
    return 0;
}