POJ 3450--->Corporate Identity(后缀数组求多个字符串的公共子串）

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

这题大概就是给你一堆字符串找到这堆字符串的公共部分

这题标准答案是kmp,我有一篇用KMP做的–》http://blog.csdn.net/my_stage/article/details/52004770

必须感叹一下，即使有后缀数组这样的数据结构但在这道题上依然还是比不过Kmp算法，同样这题也是后缀数组的一个模板题，希望大家可以有所收获，

首先这题很其他模板题一样，先是把所以字符串连接，之后每个字符串中间用一个从未出现的字符分隔，比如这题中我把所以的小写转换成数字，那么我所用的分隔数是30，那么每次输入一个新的字符都会让这个分隔数加1，这样便于我们构建后缀数组，并且我们用一个数组标记输入字符串的序号。这是为了之后的操作做的准备。
之后通过2倍倍增算法（也就是那边传说中的论文里的模板算法）求出sa,rank,height数组。
之后我们将答案二分，记住 left=1,right=len(len为最后一个字符串的长度） 之后开始二分，记住不要让left=0,因为这样最后会得到mid=0之后让我们无法判断答案是否存在。

具体请看代码：

#include <stdio.h>
#include <algorithm>
#include <string.h>
#define maxs 1000005
#include <iostream>
using namespace std;
#define MME(i,j) memset(i,j,sizeof(i))
int s[maxs],sa[maxs],rank[maxs],height[maxs];
int wa[maxs],loc[maxs],wb[maxs],wv[maxs],wd[maxs];
char input[205];
int ans;
int cmp(int *r,int a,int b,int L)
{
    return r[a]==r[b]&&r[a+L]==r[b+L];
}

void get_sa(int *r,int n,int m)
{
    int i,j,p,*x=wa,*y=wb;
    for(i=0;i<m;i++) wd[i]=0;
    for(i=0;i<n;i++) wd[x[i]=r[i]]++;
    for(i=1;i<m;i++) wd[i] +=wd[i-1];
    for(i=n-1;i>=0;i--) sa[--wd[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++) wd[i] = 0;

        for(i=0;i<n;i++) wd[wv[i]]++;
        for(i=1;i<m;i++) wd[i]+=wd[i-1];

        for(i=n-1;i>=0;i--) sa[--wd[wv[i]]] = y[i];

        for(swap(x,y),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++;
        }
    }
/*
    for(int i=0;i<n;i++)
        printf("SA[%d] is %d\n",i,sa[i]);
*/}

void build_height(int *r,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rank[sa[i]]= i;
    for(i=0;i<n;height[rank[i++]]=k){

        for(k ? k-- : 0, j=sa[rank[i]-1]; r[i+k] == r[j+k] ; k++);
    }
/*
    for(int i=0;i<=n;i++)
        printf("h[%d] is %d\n",i,height[i]);
*/}
bool vis[4010];
char ans_str[maxs];
bool check(int mid,int len,int appear)
{
     MME(vis,0);//len 为 s 数组 的长度
     int i,j,tmp=0;
     for(i=2;i<len;i++)
     {
         if(height[i]<mid)
         {
             tmp=0;
             MME(vis,0);
             continue;
         }
         if(!vis[loc[sa[i-1]]])
         {
             tmp++;
             vis[loc[sa[i-1]]]=1;
         }
         if(!vis[loc[sa[i]]])
         {
             tmp++;
             vis[loc[sa[i]]]=1;
         }
         if(tmp==appear)
         {
            //printf("MId is %d    I is %d\n",mid,i);
             for(j=0;j<mid;j++)
             {
                    ans_str[j]=s[sa[i]+j]+'a'-2;
             }

            // ans_str[mid]='\0';
             return true;
         }
            //printf("MID is %d\n",mid);
            ans_str[mid]='\0';
     }
     return 0;
}


int main()
{
    int cnt,n,len;
    while(~scanf("%d",&n)&&n)
    {
        ans=0;
        cnt=0;
        int j,id,temp=30;
        for(int i=0;i<n;i++)
        {
           // memset(input,'\0',sizeof(input));
            scanf("%s",input);
            len=strlen(input);

            for(j=0;j<len;j++)
            {
               id = input[j]-'a'+2;
               loc[cnt] = i; //  把字符串分组
               s[cnt++] = id;
            }
            loc[cnt] =temp;
            s[cnt++] =temp++;
        }
        s[cnt]=0;

        get_sa(s,cnt+1,temp);
        build_height(s,cnt);

        int left=1,right=len,mid;
        bool flag=0;

        while(left<=right)
        {
            mid=(left+right)/2;
            if(check(mid,cnt,n))
            {
                flag=1;
                left=mid+1;
             //   ans=mid;
            }
            else
                right=mid-1;
        }
        if(flag)
            puts(ans_str);
        else printf("IDENTITY LOST\n");

    }
    return 0;
}