POJ 3294 Life Forms <后缀数组+二分>

题目：传送门。

分析：求重复出现大于n/2次的最长子串，有多个时按字典序输出。对长度进行二分。RE了几次，注意满足条件的子串有可能有超过1000个。

代码：

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <string>
#include <cstring> 
using namespace std;

const int MAXN=110;
const int MAXL=1010;
int wm[MAXN*MAXL],wa[MAXN*MAXL],wb[MAXN*MAXL],height[MAXN*MAXL],sa[MAXN*MAXL];
int *rank;

bool cmp(int *r,int a,int b,int len){
	return r[a]==r[b]&&r[a+len]==r[b+len];
}

void da(int *data,int *sa,int n,int m){
	int *x=wa,*y=wb,*t,i,j,p;
	for(i=0;i<m;++i) wm[i]=0;
	for(i=0;i<n;++i) ++wm[x[i]=data[i]];
	for(i=1;i<m;++i) wm[i]+=wm[i-1];
	for(i=n-1;i>=0;--i) sa[--wm[x[i]]]=i;
	for(j=p=1;p<n;j<<=1,m=p){
		for(p=0,i=n-j;i<n;++i) y[p++]=i;
		for(i=0;i<n;++i) if(sa[i]>=j) y[p++]=sa[i]-j;
		for(i=0;i<m;++i) wm[i]=0;
		for(i=0;i<n;++i) ++wm[x[y[i]]];
		for(i=1;i<m;++i) wm[i]+=wm[i-1];
		for(i=n-1;i>=0;--i) sa[--wm[x[y[i]]]]=y[i];
		for(t=x,x=y,y=t,i=p=1,x[sa[0]]=0;i<n;++i) x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
	}
	rank=x;
}
 
void calHeight(int *data,int *sa,int n){
 	for(int i=0,j,k=0;i<n;height[rank[i++]]=k)
 		for(k?--k:k,j=sa[rank[i]-1];data[i+k]==data[j+k];++k);
}

int len[MAXN],data[MAXL*MAXN];
int n,a[MAXL],t,cas=0;
char str[MAXL];
int nn;
bool vis[MAXN];

bool judge(int _len,int x){
	t=0;
	int flag,counte=0;
	for(int i=1;i<=_len;++i){
		if(height[i]>=x){
			int posi1=sa[i-1],posi2=sa[i];
			if(!(flag++)) memset(vis,false,sizeof vis);
			for(int j=1;j<=n;++j){
				if(posi1>=len[j-1]&&posi1<len[j]) counte+=vis[j-1]?0:1,vis[j-1]=true;
				if(posi2>=len[j-1]&&posi2<len[j]) counte+=vis[j-1]?0:1,vis[j-1]=true;
				if(counte>n/2) break;
			}
		}
		else{
			if(counte>n/2) a[t++]=i-1;
			flag=counte=0;
		}
	}
	if(counte>n/2) a[t++]=_len;
	if(t){
		nn=t;
		return true;
	}
	else return false;
}

void solve(int _len){
	int l=0,r=MAXL;
	while(l<r){
		int mid=l+(r-l+1)/2;
		if(judge(_len,mid)) l=mid;
		else r=mid-1;
	}
	if(!l) cout<<"?\n";
	else{
		for(int i=0;i<nn;++i){
			int star=sa[a[i]];
			for(int j=star;j<star+l;++j){
				printf("%c",data[j]);
			}
			printf("\n");
		}
	}
}

int main(){
	int p;
	while(cin>>n,n){
		p=len[0]=0;
		for(int i=0;i<n;++i){
			scanf("%s",str);
			int temp_len=strlen(str);
			len[i+1]=len[i]+temp_len+1;
			for(int j=0;j<temp_len;++j){
				data[p++]=str[j];
			}
			data[p++]=130+i;
		}
		if(cas++) cout<<endl;
		if(n==1){
			printf("%s\n",str);
			continue;
		}
		data[p-1]=0;
		da(data,sa,p,300);
		calHeight(data,sa,p-1);
		solve(p-1);
	}
	return 0;
}