【POJ1226】Substrings（后缀数组，二分）

题意：

n<=10,len<=100

思路：

只有一个字符串的时候特判一下

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
#define fi first
#define se second
#define MP make_pair
#define N   210000
#define MOD 1000000007
#define eps 1e-8
#define pi acos(-1)
#define oo  1000000000
 
char ch[N];
 
int n,i,s[N],sa[N],wa[N],wb[N],wc[N],wd[N],height[N],rank[N],
    a[N],b[N],num[N],M;
 
int read()
{
   int v=,f=;
   char c=getchar();
   while(c<||<c) {if(c=='-') f=-; c=getchar();}
   while(<=c&&c<=) v=(v<<)+v+v+c-,c=getchar();
   return v*f;
}
 
bool cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
 
void getsa(int *r,int *sa,int n,int m)
{
    int *x=wa,*y=wb,j,p;
    for(i=;i<n;i++) wc[x[i]=r[i]]++;
    for(i=;i<m;i++) wc[i]+=wc[i-];
    for(i=n-;i>=;i--) sa[--wc[x[i]]]=i;
    for(j=,p=;p<n;j*=,m=p)
    {
        p=;
        for(i=n-j;i<n;i++) y[p++]=i;
        for(i=;i<n;i++)
         if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=;i<n;i++) wd[i]=x[y[i]];
        for(i=;i<m;i++) wc[i]=;
        for(i=;i<n;i++) wc[wd[i]]++;
        for(i=;i<m;i++) wc[i]+=wc[i-];
        for(i=n-;i>=;i--) sa[--wc[wd[i]]]=y[i];
        swap(x,y);
        p=; x[sa[]]=;
        for(i=;i<n;i++) x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
    }
}
 
void getheight(int *r,int *sa,int n)
{
    int i,j,k=;
    for(i=;i<=n;i++) rank[sa[i]]=i;
    for(i=;i<n;height[rank[i++]]=k)
    {
        if(k) k--;
        j=sa[rank[i]-];
        while(r[i+k]==r[j+k]) k++;
    }
}
 
void init()
{
        memset(s,,sizeof(s));
        memset(sa,,sizeof(sa));
        memset(wa,,sizeof(wa));
        memset(wb,,sizeof(wb));
        memset(wc,,sizeof(wc));
        memset(wd,,sizeof(wd));
        memset(height,,sizeof(height));
        memset(rank,,sizeof(rank));
}
 
bool isok(int K)
{
    int i=;
    while(i<n)
    {
        i++;
        if(height[i]>=K)
        {
            int st=i;
            while(i<=n&&height[i]>=K) i++;
            for(int j=;j<=M;j++) b[j]=;
            for(int j=max(,st-);j<=i-;j++)
            {
                int t=sa[j];
                b[num[t]]++;
            }
            int flag=;
            for(int j=;j<=M;j++)
             if(!b[j]){flag=; break;}
            if(flag) return ;
        }
    }
    return ;
}
 
int main()
{
    //freopen("poj1226.in","r",stdin);
    //freopen("poj1226.out","w",stdout);
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        init();
        scanf("%d",&M);
        n=-;
        int p=;
        for(int i=;i<=M;i++)
        {
            scanf("%s",ch);
            int t=strlen(ch);
            if(M==) printf("%d\n",t);
            for(int j=;j<t;j++)
            {
                s[++n]=ch[j]+;
                num[n]=i;
            }
            s[++n]=p++;
            num[n]=;
            for(int j=t-;j>=;j--)
            {
                s[++n]=ch[j]+;
                num[n]=i;
            }
            s[++n]=p++;
            num[n]=;
        }
        if(M==) continue;
        //printf("1\n");
        getsa(s,sa,n+,);
        getheight(s,sa,n);
        //for(int i=2;i<=n;i++) printf("%d\n",height[i]);
        int L=;
        int R=;
        int last=;
        while(L<=R)
        {
            int mid=(L+R)>>;
            if(isok(mid)){last=mid; L=mid+;}
             else R=mid-;
        }
        printf("%d\n",last);
        memset(num,,sizeof(num));
    }
    return ;
}