POJ - 3693 求这串字符中的重复次数最多的连续重复子串,多组答案输出字典序最小的那个串(后缀数组)

题目:求这串字符中的重复次数最多的连续重复子串,多组答案输出字典序最小的那个串。

思路:与前一个题目几乎一样的,加上了字典序。多判断就好

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100010;
int t1[maxn],t2[maxn],c[maxn];
bool cmp(int *r,int a,int b,int l)
{
	return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int str[],int sa[],int ra[],int height[],int n,int m)
{
	n++;
	int p,*x=t1,*y=t2;
	for(int i=0;i<m;i++) c[i]=0;
	for(int i=0;i<n;i++) c[x[i]=str[i]]++;
	for(int i=1;i<m;i++) c[i]+=c[i-1];
	for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
	for(int j=1;j<=n;j<<=1)
	{
		p=0;
		for(int i=n-j;i<n;i++) y[p++]=i;
		for(int i=0;i<n;i++)
		if(sa[i]>=j)  y[p++]=sa[i]-j;
		for(int i=0;i<m;i++) c[i]=0;
		for(int i=0;i<n;i++) c[x[y[i]]]++;
		for(int i=1;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]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
		if(p>=n) break;
		m=p;
	}
	int k=0;
	n--;
	for(int i=0;i<=n;i++) ra[sa[i]]=i;
	for(int i=0;i<n;i++)
	{
		if(k) k--;
		int j=sa[ra[i]-1];
		while(str[i+k]==str[j+k])k++;
		height[ra[i]]=k;
	}
}

int m,n,k,t;
int ra[maxn],height[maxn],str[maxn],sa[maxn];

int mm[maxn],best[20][maxn];
void initRMQ(int n)
{
    mm[0]=-1;
    for(int i=1;i<=n;i++)
        mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
    for(int i=1;i<=n;i++)best[0][i]=height[i];
    for(int i=1;i<=mm[n];i++)
        for(int j=1;j+(1<<i)-1<=n;j++)
        {
            int a=best[i-1][j];
            int b=best[i-1][j+(1<<(i-1))];
            if(a<b) best[i][j]=a;
            else best[i][j]=b;
        }
}
int askRMQ(int a,int b)
{
    int t=mm[b-a+1];
    b-=(1<<t)-1;
    a=best[t][a];b=best[t][b];
    return a<b?a:b;
}
int lcp(int a,int b)
{
    a=ra[a];b=ra[b];
    if(a>b) swap(a,b);
    return askRMQ(a+1,b);
}

char s[maxn];
int main()
{
    ///freopen("in.txt","r",stdin);
    int cas=0;
    while(scanf("%s",s))
    {
        if(s[0]=='#') break;
        n=strlen(s);
        for(int i=0;i<n;i++)
        str[i]=s[i]-'a'+1;
        str[n]=0;
        da(str,sa,ra,height,n,28);
/*
        for(int i=1; i<=n; ++i)
		{
		      printf("%d:\t",i);
		      for(int j=i-1; j<n; ++j)
		            printf("%d",str[j]);
		      puts("");
		}
		puts("");
		puts("-------------After sort---------------");
		for(int i=1; i<=n; ++i)
		{
		      printf("sa[%2d ] = %2d\t",i,sa[i]);
		      for(int j=sa[i]; j<n; ++j)//for(int j=sa[i]; j<n; ++j)???
		            printf("%d",str[j]);
		      puts("");
		}
		puts("");
		puts("---------------Height-----------------");
		for(int i=1; i<=n; ++i)
		      printf("height[%2d ]=%2d \n",i,height[i]);
		puts("");
		puts("----------------ra------------------");
		for(int i=1; i<=n; ++i)
		      printf("ra[%2d ] = %2d\n",i,ra[i]);
		puts("------------------END-----------------");

*/
        initRMQ(n);
        int ans=1,l=0,r=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j+i<n;j+=i)//下面的代码上图都有体现
            {
                int len=lcp(j,j+i);
                int k=j-(i-len%i);
                int sum=len/i+1;

                //if(k>=0&&lcp(k,k+i)>=i)
                //    sum++;
                int pl=j;
                for(int m=j+len%i;m>j-i&&m>=0&&str[m]==str[m+i];m--)//找字典序最小的 l 
                {
                    if(m==k)
                    {
                        sum++;
                        pl=m;
                    }
                    else if(ra[m]<ra[pl])
                    {
                        pl=m;
                    }
                }
                if(ans<sum)
                {
                    ans=sum;
                    l=pl;
                    r=l+sum*i-1;
                }
                else if(ans==sum&&ra[l]>ra[pl])
                {
                    l=pl;
                    r=l+sum*i-1;
                }
            }
        }
        printf("Case %d: ",++cas);
        for(int i=l;i<=r;i++)
            printf("%c",s[i]);
        puts("");
    }
	return 0;
}

猜你喜欢

转载自blog.csdn.net/dllpxfire/article/details/81253738