spoj220 后缀数组+二分

版权声明:欢迎转载,转载请注明出处,如有错误,还望指出,谢谢 博客地址:https://blog.csdn.net/lanyanzhiji123asd https://blog.csdn.net/lanyanzhiji123asd/article/details/89598974
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<set>
#include<queue>
using namespace std;
#define ll long long
int n,m;
const int ding=1e5+100;
const int INF=0x3f3f3f3f;
int sa[ding],x[ding],y[ding],c[ding],r[ding],strlenn[ding],rankk[ding],height[ding],flag=0,vis[120],ans[ding];
int minn[ding],maxn[ding];
int cas=1;
string str;
void get_SA()
{
	int i,k;
//	memset(x,0,sizeof(x));//因为我是以a来作为终止符,所以后面的空 应该设置成-1 
//	memset(y,0,sizeof(y));
	memset(c,0,sizeof(c));//清空桶;
	for(i=0;i<n;i++)	c[x[i]=r[i]]++; 
	for(i=1;i<m;i++)	c[i]+=c[i-1];
	for(i=n-1;i>=0;i--)	
	{
		sa[--c[x[i]]]=i;
	}
	int p=1;
	for(k=1;k<n;k<<=1,m=p)
	{
		p=0;
		for(i=n-k;i<n;i++)	y[p++]=i;//
		for(i=0;i<n;i++)	if(sa[i]>=k) y[p++]=sa[i]-k;//大于等于k 
		//sa是第一关键字的排名, 然后再根据第二关键字的排名来确定排名, 
		memset(c,0,sizeof(c));
		for(i=0;i<n;i++)	c[x[y[i]]]++;//统计当前字符串 
		for(i=1;i<m;i++)	c[i]+=c[i-1];//统计个数 
		for(i=n-1;i>=0;i--)		sa[--c[x[y[i]]]]=y[i];     //重置sa数组
		memcpy(y,x,sizeof(x));
		memset(x,0,sizeof(x));
//		swap(x,y);
		p=1;//从1开始编号 
		x[sa[0]]=0;
		//因为上面已经重置了x的值  所以下面是从1开始循环,然后是和前一个比较sa[i]和sa[i-1] 比较 
		for(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++;
		
		//x数组里面记住的是x[i]=j  下标为i的后缀,排名为j
		//y数组的内容是y[i]=j 下标为j 的后缀排名为i 
		if(p>=n)break;
	}
//	for(i=0;i<n;i++)
//		cout<<sa[i]<<" ";
//	cout<<endl;
}

void get_height()
{
	int i,k=0,j;
	for(i=0;i<n;i++) rankk[sa[i]]=i;
	for(i=0;i<n-1;i++)
	{//为什么循环到n-1,因为我们在原始字符串后面加了a,所以这个后缀一定是最小的。 

		if(k)k--;
		j=sa[rankk[i]-1];
		for(;r[i+k]==r[j+k];k++);
		height[rankk[i]]=k; 
	}
//	for(i=0;i<n-1;i++)
//		cout<<height[i]<<" ";
//	cout<<endl;
		
} 

int check(int mid,int kk)
{
	int cnt=0,i,temp,temp1;
	memset(vis,0,sizeof(vis));
	memset(minn,INF,sizeof(minn));
	memset(maxn,-INF,sizeof(maxn));
	for(i=2;i<n;i++)
	{
		if(height[i]>=mid)
		{
			temp=strlenn[sa[i]];
			temp1=strlenn[sa[i-1]];
			maxn[temp]=max(maxn[temp],sa[i]);
			minn[temp]=min(minn[temp],sa[i]);
			maxn[temp1]=max(maxn[temp1],sa[i-1]);
			minn[temp1]=min(minn[temp1],sa[i-1]);
			if(!vis[temp]&&maxn[temp]-minn[temp]>=mid)
			{
				cnt++;
				vis[temp]=1;
			}
			if(!vis[temp1]&&maxn[temp1]-minn[temp1]>=mid)
			{
				cnt++;
				vis[temp1]=1;
			}
			
		}
		else
		{
			if(cnt==kk)
			{
				return 1;
			}
			cnt=0;
			memset(vis,0,sizeof(vis));
			memset(minn,INF,sizeof(minn));
		memset(maxn,-INF,sizeof(maxn));
		}
	}
	if(cnt==kk)
	{
		return 1;
	}
	cnt=0;
	return 0;
}

int main()
{
 	std::ios::sync_with_stdio(false);
	int t,len,ans,kk,i,j,L,R;
	cin>>t;
	string tol,str;
	while(t--)
	{
		n=0,m=300;
		tol="";
		cin>>kk;
		len=0;
		ans=0;
		for(i=1;i<=kk;i++)
		{
			cin>>str;
//			cout<<str<<endl;
			tol+=str;
			len=max(len,(int)str.length());
			tol+=(char)('$'+i);
			for(j=0;j<str.length();j++)
			{
				strlenn[n]=i;
				r[n++]=(int)(str[j]);
			}
			strlenn[n]=i;
			r[n++]=(299-i);	
		}
		r[n-1]=0;
//		for(i=0;i<n;i++)
//			cout<<"r   "<<r[i]<<endl;
		get_SA();
		get_height();
		L=0;R=len;
		while(L<=R)
		{
			int mid=(L+R)/2;
			if(check(mid,kk))
			{
				ans=mid;
				L=mid+1;
			}
			else R=mid-1;
		}
		cout<<ans<<endl;
	}

 return 0;
}

猜你喜欢

转载自blog.csdn.net/lanyanzhiji123asd/article/details/89598974
今日推荐