POJ - 1226 Substrings(后缀数组+二分)

题目链接：点击查看

题目大意：给出n个字符串，求出n个字符串中或者在他们翻转过来的字符串中，出现的最长公共子串的长度

题目分析：说白了就是求n个字符串中的最长公共子串的变形，而这个变形也相当简单，只是需要将其翻转后的字符串一起拼接起来然后二分就是答案了

注意特判n==1的情况，因为忘记特判，WA了好久。。

代码：

#include<iostream>
#include<cstdio> 
#include<string>
#include<ctime>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<cmath>
#include<sstream>
using namespace std;
 
typedef long long LL;
 
const int inf=0x3f3f3f3f;

const int N=1e5+100;
 
char str[N];
 
int sa[N]; //SA数组，表示将S的n个后缀从小到大排序后把排好序的
//的后缀的开头位置顺次放入SA中
int t1[N],t2[N],c[N];
 
int rk[N],height[N],belong[N],len,n,sum;
 
int s[N];

bool vis[110];
 
void build_sa(int s[],int n,int m)//n为添加0后的总长
{
    int i,j,p,*x=t1,*y=t2;
    for(i=0;i<m;i++) 
		c[i]=0;
    for(i=0;i<n;i++) 
		c[x[i]=s[i]]++;
    for(i=1;i<m;i++) 
		c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) 
		sa[--c[x[i]]]=i;
    for(j=1;j<=n;j<<=1) 
	{
        p=0;
        for(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++) 
			c[i]=0;
        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];
        swap(x,y);
        p=1,x[sa[0]]=0;
        for(i=1;i<n;i++) 
            x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++;
        if(p>=n) 
			break;
        m=p;
    }
}
 
void get_height(int s[],int n)//n为添加0后的总长
{
    int i,j,k=0;
    for(i=0;i<=n;i++)
        rk[sa[i]]=i;
    for(i=0;i<n;i++) 
	{
        if(k) 
			k--;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) 
			k++;
        height[rk[i]]=k;
    }
}
 
void solve(int base=128)
{
	build_sa(s,len+1,base);
	get_height(s,len);
}

bool check(int mid)
{
	memset(vis,false,sizeof(vis));
	for(int i=1;i<=len;i++)
	{
		if(!belong[sa[i]])
			break;
		if(height[i]<mid)
		{
			memset(vis,false,sizeof(vis));
			vis[belong[sa[i]]]=true;
			sum=1;
		}
		else
		{
			if(!vis[belong[sa[i]]])
			{
				vis[belong[sa[i]]]=true;
				sum++;
			}
		}
		if(sum==n)
			return true;
	}
	return false;
}

int main()
{
//	freopen("input.txt","r",stdin);
//	ios::sync_with_stdio(false);
	int w;
	cin>>w;
	while(w--)
	{
		memset(belong,0,sizeof(belong));
		scanf("%d",&n);
		if(n==1)
		{
			scanf("%s",str);
			printf("%d\n",strlen(str));
			continue;
		}
		int pos=130,tot=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%s",str);
			for(int j=0;str[j];j++)
			{
				s[tot]=str[j];
				belong[tot]=i;
				tot++;
			}
			s[tot++]=pos++;
			for(int j=strlen(str)-1;j>=0;j--)
			{
				s[tot]=str[j];
				belong[tot]=i;
				tot++;
			}
			s[tot++]=pos++;
		}
		len=tot;
		s[len]=0;
		solve(pos);
		int l=0,r=inf,ans;
		while(l<=r)
		{
			int mid=l+r>>1;
			if(check(mid))
			{
				ans=mid;
				l=mid+1;
			}
			else
				r=mid-1;
		}
		printf("%d\n",ans);
	}
	
	
	
	
	
	
	
	
	
	
	
	
	
	
	return 0;
}