poj 1226 Substrings( 后缀数组 出现或反转后出现在每个字符串中的最长子串)

题意 : 给定 n 个字符串，求出现或反转后出现在每个字符串中的最长子串。

思路 : 讲n个字符串以及这n个字符串反转之后的串用没有出现过的字符串拼接起来，然后求height，二分答案L，再分组，如果有一组中包含了所有字符串( 或者其反转的串 ) ，那么长度L可行。

#include <stdio.h>  
#include <string.h>
#include <string>
#include <vector>
#include <iostream>
#include <algorithm>  
using namespace std;  

#define maxn 15005  
int wa[maxn],wb[maxn],wv[maxn],wt[maxn];  

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

void da(int *r,int *sa,int n,int m){  
	int i,j,p,*x=wa,*y=wb,*t;  
	for(i=0;i<m;i++) wt[i]=0;  
	for(i=0;i<n;i++) wt[x[i]=r[i]]++;  
	for(i=1;i<m;i++) wt[i]+=wt[i-1];  
	for(i=n-1;i>=0;i--) sa[--wt[x[i]]]=i;  
	for(j=1,p=1;p<n;j*=2,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<n;i++) wv[i]=x[y[i]];  
		for(i=0;i<m;i++) wt[i]=0;  
		for(i=0;i<n;i++) wt[wv[i]]++;  
		for(i=1;i<m;i++) wt[i]+=wt[i-1];  
		for(i=n-1;i>=0;i--) sa[--wt[wv[i]]]=y[i];  
		for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)  
			x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;  
	}  
}  

int Rank[maxn],height[maxn];  

void calheight(int *r,int *sa,int n){  
	int i , j , k = 0 ;  
	for( i=1 ; i<=n ; i++ ) Rank[sa[i]]=i;  
	for(i=0;i<n;i++) {  
		if(k)k--;  
		int j = sa[Rank[i]-1];  
		while(r[i+k]==r[j+k]) k++ ;  
		height[Rank[i]] = k ;  
	}     
	return;  
}  

int r[maxn] , sa[maxn] ;  
char str[maxn] ;
// in[i] 表示第 i 个字符属于第几个字符串
int in[maxn] ;
// have 用于辅助计算每组内属于不同字符串的个数
bool have[205] ;
int n ;

bool check( int limit , int len ) {
	memset( have , false , sizeof(have) ) ;
	if( in[sa[1]] ) {
		have[in[sa[1]]] = true ;
	}
	for( int i = 2 ; i <= len ; i ++ ) {
		if( height[i] < limit ) {
			int cnt = 0 ;
			for( int j = 1 ; j <= n ; j ++ ) {
				if( have[j] ) cnt ++ ;
			}
			if( cnt == n ) return true ;
			memset( have , false , sizeof(have) ) ;	
		}
		if( in[sa[i]] ) {
			have[in[sa[i]]] = true ;
		}
	}
	int cnt = 0 ;
	for( int j = 1 ; j <= n ; j ++ ) {
		if( have[j] ) cnt ++ ;
	}
	return cnt == n ;
}
int main(){
	int cas ;
	scanf( "%d" , &cas ) ;
	while( cas -- ) {
		scanf( "%d" , &n ) ;
		int len = 0 ;
		memset( in , 0 , sizeof(in) ) ;
		int mm = 99999999 ;
		for( int i = 1 ; i <= n;  i ++ ) {
			scanf( "%s" , str ) ;
			int l = strlen( str ) ;
			mm = min( mm , l ) ;
			for( int j = 0 ; j < l ; j ++ ) {
				r[len] = str[j] ;
				in[len++] = i ;
			}
			r[len++] = 'z' + i ;
			for( int j = l - 1 ; j >= 0 ; j -- ) {
				r[len] = str[j] ;
				in[len++] = i ;
			}
			r[len++] = 'z' + i + n ;
		}
		r[len] = 0 ;
		da( r , sa , len + 1 , 400 ) ;
		calheight( r , sa , len ) ;
		int l = 0 , r = mm , m ; 
		while( l < r ) {
			m = ( l + r + 1 ) >> 1 ;
			if( check( m , len ) ) {
				l = m ;
			}else{
				r = m - 1 ;
			}
		}
		printf( "%d\n" , l ) ;
	}
	return 0 ;
}