【后缀数组】【poj 3294】Life Forms

http://poj.org/problem?id=3294

hehe被虐

基本应用——不小于k个字符串的最长子串

一周没写果然手生了

//#define _TEST _TEST
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
/************************************************
Code By willinglive    Blog:http://willinglive.cf
************************************************/
#define rep(i,l,r) for(int i=(l),___t=(r);i<=___t;i++)
#define per(i,r,l) for(int i=(r),___t=(l);i>=___t;i--)
#define MS(arr,x) memset(arr,x,sizeof(arr))
#define LL long long
#define INE(i,u,e) for(int i=head[u];~i;i=e[i].next)
inline const int read()
{int r=0,k=1;char c=getchar();for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;
for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';return k*r;}
/
int n;
char s[100010];
int start[110],len[110],sz;
int t1[100010],t2[100010],c[100010];
int sa[100010],height[100010],rk[100010];
bool flag[110];
/
void get_sa(int *x,int *y,int *sa,int n,int m)
{
	rep(i,0,m-1) c[i]=0;
	rep(i,0,n-1) c[x[y[i]]]++;
	rep(i,1,m-1) c[i]+=c[i-1];
	per(i,n-1,0) sa[--c[x[y[i]]]]=y[i];
}
void get_height(char *s,int *sa,int *height,int n)
{
	int i,j,k=0;
	rep(i,1,n) rk[sa[i]]=i;
	for(i=0;i<n;height[rk[i++]]=k)
	    for(k?k--:0,j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
}
void da(char *s,int *sa,int *height,int n)
{
	int i,j,p,*x=t1,*y=t2,*t,m=128;
	rep(i,0,n-1) x[i]=s[i],y[i]=i;
	get_sa(x,y,sa,n,m);
	for(j=p=1;p<n;j<<=1,m=p)
	{
		p=0;
		rep(i,n-j,n-1) y[p++]=i;
		rep(i,0,n-1) if(sa[i]>=j) y[p++]=sa[i]-j;
		get_sa(x,y,sa,n,m);
		t=x;x=y;y=t;p=1;x[sa[0]]=0;
		rep(i,1,n-1) x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]?p-1:p++;
	}
	get_height(s,sa,height,n);
}
bool check(int md)
{
	MS(flag,0);
	int cnt=0;
	rep(i,2,sz)
	{
        if(height[i]>=md)
		{
			rep(j,1,n)
			{
				if(start[j]<=sa[i] && sa[i]+md-1<start[j]+len[j])
				{
					if(!flag[j]) flag[j]=1,cnt++;
				}
				if(start[j]<=sa[i-1] && sa[i-1]+md-1<start[j]+len[j])
				{
					if(!flag[j]) flag[j]=1,cnt++;
				}
			}
		}
		else
		{
			if(cnt>n/2) return 1;
			cnt=0;
			MS(flag,0);
		}
	}
	return cnt>n/2;
}
void cal(int md)
{
	MS(flag,0);
	int cnt=0;
	rep(i,2,sz)
	{
        if(height[i]>=md)
		{
			rep(j,1,n)
			{
				if(start[j]<=sa[i] && sa[i]+md-1<start[j]+len[j])
				{
					if(!flag[j]) flag[j]=1,cnt++;
				}
				if(start[j]<=sa[i-1] && sa[i-1]+md-1<start[j]+len[j])
				{
					if(!flag[j]) flag[j]=1,cnt++;
				}
			}
		}
		else
		{
			if(cnt>n/2)
			{
				rep(j,sa[i-1],sa[i-1]+md-1) putchar(s[j]);
				puts("");
			}
			cnt=0;
			MS(flag,0);
		}
	}
	if(height[sz]<md) cnt=0;
	if(cnt>n/2)
	{
		rep(j,sa[sz-1],sa[sz-1]+md-1) putchar(s[j]);
		puts("");
	}
}
/
void input()
{
	MS(sa,0); MS(height,0); MS(rk,0); MS(t1,0); MS(t2,0); MS(c,0);MS(start,0); MS(len,0); sz=0;
	
	sz=1;
    rep(i,1,n)
    {
    	scanf("%s",s+sz);
    	start[i]=sz;
    	sz+=len[i]=strlen(s+sz);
    }
    sz--;
}
void solve()
{
	da(s,sa,height,sz+1);
	int l=0,r=sz/n+1,mid;
	while(l<r)
	{
		mid=l+r+1>>1;
		if(check(mid)) l=mid;
		else r=mid-1;
	}
	if(!l) puts("?");
	else cal(l);
	puts("");
}
/
int main()
{
    #ifndef _TEST
    freopen("std.in","r",stdin); freopen("std.out","w",stdout);
    #endif
    while(n=read(),n)
    input(),solve();
    return 0;
}