poj3294 后缀数组emmm

版权声明:欢迎转载,转载请注明出处,如有错误,还望指出,谢谢 博客地址:https://blog.csdn.net/lanyanzhiji123asd https://blog.csdn.net/lanyanzhiji123asd/article/details/89577181

TLE一下午,memset不能用太多,吧memset的数组大小设置成正好,别设置太大

#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,kk;
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[1200],ans[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 t)
{
	int size=0,cnt=0,i,j;
//	for(i=1;i<=t;i++)
//		vis[i]=0;
	memset(vis,0,sizeof(vis));
	for(i=2;i<n;i++)
	{//cout<<sa[i-1]<<"   sa  "<<sa[i]<<" "<<height[i]<<endl;
		if(height[i]>=mid)
		{
			int temp=strlenn[sa[i]];
			int temp1=strlenn[sa[i-1]];
			if(!vis[temp])
			{//cout<<"qqqqqq"<<endl;
				cnt++;
				vis[temp]=1;
			}
			if(!vis[temp1])
			{//cout<<"ttttttttt"<<endl;
				cnt++;
				vis[temp1]=1;
			}
		}
		else
		{//cout<<"uuuuuuuuuuu"<<endl;
			if(cnt>t/2)
			{
				flag=1; 
//				cout<<i<<" aaaa "<<sa[i-1]<<endl;
				ans[++size]=sa[i-1];
			}
			cnt=0;
//			for(i=1;i<=t;i++)
//				vis[i]=0;
			memset(vis,0,sizeof(vis));
		}
	}
	if(cnt>t/2)
	{
		flag=1; 
//		cout<<i<<" bbbb "<<sa[i-1]<<endl;
		ans[++size]=sa[i-1];
		
	}
	cnt=0;
//	for(i=1;i<=t;i++)
//		vis[i]=0;
	memset(vis,0,sizeof(vis));
	if(size)
	{
		ans[0]=size;
		return 1;
	}
	return 0;
}

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

 return 0;
}

猜你喜欢

转载自blog.csdn.net/lanyanzhiji123asd/article/details/89577181