【poj1226-出现或反转后出现在每个串的最长公共子串】后缀数组

题意：求n个串的最长公共子串，子串出现在一个串中可以是它的反转串出现。总长<=10^4.

题解：

对于每个串，把反转串也连进去。
二分长度，分组，判断每个组。

 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;

const int N=2*21000;
int n,sl,cl,c[N],rk[N],sa[N],Rs[N],wr[N],y[N],h[N],st[N],ed[N],fst[N],fed[N];
char s[N];
bool vis[N];

void get_sa(int m)
{
    for(int i=1;i<=cl;i++) rk[i]=c[i];
    for(int i=1;i<=m;i++) Rs[i]=0;
    for(int i=1;i<=cl;i++) Rs[rk[i]]++;
    for(int i=1;i<=m;i++) Rs[i]+=Rs[i-1];
    for(int i=cl;i>=1;i--) sa[Rs[rk[i]]--]=i;
    
    int ln=1,p=0;
    while(p<cl)
    {
        int k=0;
        for(int i=cl-ln+1;i<=cl;i++) y[++k]=i;
        for(int i=1;i<=cl;i++) if(sa[i]>ln) y[++k]=sa[i]-ln;
        
        for(int i=1;i<=cl;i++) wr[i]=rk[y[i]];
        for(int i=1;i<=m;i++) Rs[i]=0;
        for(int i=1;i<=cl;i++) Rs[wr[i]]++;
        for(int i=1;i<=m;i++) Rs[i]+=Rs[i-1];
        for(int i=cl;i>=1;i--) sa[Rs[wr[i]]--]=y[i];
        
        for(int i=1;i<=cl;i++) wr[i]=rk[i];
        for(int i=cl+1;i<=cl+ln;i++) wr[i]=0;
        p=1;rk[sa[1]]=1;
        for(int i=2;i<=cl;i++)
        {
            if(wr[sa[i]]!=wr[sa[i-1]] || wr[sa[i]+ln]!=wr[sa[i-1]+ln]) p++;
            rk[sa[i]]=p;
        }
        ln*=2,m=p;
    }
    sa[0]=0,rk[0]=0;
}

void get_h()
{
    int k=0,j;
    for(int i=1;i<=cl;i++) if(rk[i]!=1)
    {
        j=sa[rk[i]-1];
        if(k) k--;
        while(c[i+k]==c[j+k] && i+k<=cl && j+k<=cl) k++;
        h[rk[i]]=k;
    }
    h[0]=0;
}

int idx(int x)
{
    for(int i=1;i<=n;i++) 
        if((st[i]<=x && x<=ed[i]) || (fst[i]<=x && x<=fed[i]))  return i;
}

bool check(int k)
{
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=cl;i++)
    {
        if(h[i]<k)
        {
            int cnt=0;
            for(int j=1;j<=n;j++) if(vis[j]) cnt++;
            if(cnt==n) return 1;
            memset(vis,0,sizeof(vis));
        }
        vis[idx(sa[i])]++;
    }
    int cnt=0;
    for(int j=1;j<=n;j++) if(vis[j]) cnt++;
    if(cnt==n) return 1;
    return 0;
}

int main()
{
    freopen("a.in","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        int num=123;
        cl=0;
        for(int i=1;i<=n;i++) 
        {
            scanf("%s",s+1);
            sl=strlen(s+1);
            if(i>1) c[++cl]=++num;
            st[i]=cl+1;for(int j=1;j<=sl;j++) c[++cl]=s[j];ed[i]=cl;
            c[++cl]=++num;
            fst[i]=cl+1;for(int j=sl;j>=1;j--) c[++cl]=s[j];fed[i]=cl;
        }
        if(n==1) {printf("%d\n",sl);continue;}
        get_sa(300);
        get_h();
        // for(int i=1;i<=cl;i++) printf("%c",c[i]);printf("\n");
        // for(int i=1;i<=cl;i++) printf("%d ",c[i]);printf("\n");
        // for(int i=1;i<=cl;i++) printf("%d ",sa[i]);printf("\n");
        // for(int i=1;i<=cl;i++) printf("%d ",rk[i]);printf("\n");
        // for(int i=1;i<=cl;i++) printf("%d ",h[i]);printf("\n");
        int l=0,r=cl,mid;
        while(l<r)
        {
            mid=(l+r+1)/2;
            if(check(mid)) l=mid;
            else r=mid-1;
        }
        printf("%d\n",l);
    }
    return 0;
}

 

这一题我曾经用kmp暴力水过。。贴一下代码

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;

const int N=110,Inf=(int)1e9;
char s[N][N],c1[N],c2[N];
int n,id,cl,l[N],nt[N],nxt[N];

void kmp_nt()
{
    nt[1]=0;
    for(int i=2,p=0;i<=cl;i++)
    {
        while(c1[p+1]!=c1[i] && p) p=nt[p];
        if(c1[p+1]==c1[i]) p++;
        nt[i]=p;
    }
    
    nxt[1]=0;
    for(int i=2,p=0;i<=cl;i++)
    {
        while(c2[p+1]!=c2[i] && p) p=nxt[p];
        if(c2[p+1]==c2[i]) p++;
        nxt[i]=p;
    }
}

bool kmp_td(int x)
{
    for(int i=1,p=0;i<=l[x];i++)
    {
        while(c1[p+1]!=s[x][i] && p) p=nt[p];
        if(c1[p+1]==s[x][i]) p++;
        if(p==cl) return 1;
    }
    for(int i=1,p=0;i<=l[x];i++)
    {
        while(c2[p+1]!=s[x][i] && p) p=nxt[p];
        if(c2[p+1]==s[x][i]) p++;
        // td[i]=p;
        if(p==cl) return 1;
    }
    return 0;
}

bool check(int len)
{
    for(int i=1;i+len-1<=l[id];i++)
    {
        cl=0;
        for(int j=i;j<=i+len-1;j++) 
            c1[++cl]=s[id][j],c2[len-cl+1]=s[id][j];
        kmp_nt();
        bool bk=1;
        for(int j=1;j<=n;j++)
            if(!kmp_td(j)) {bk=0;break;}
        if(bk) return 1;
    }
}

int main()
{
    freopen("a.in","r",stdin);
    freopen("vio.out","w",stdout);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        int ll=0,rr=Inf,mid;
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s[i]+1);
            l[i]=strlen(s[i]+1);
            if(l[i]<rr) rr=l[i],id=i;
        }
        while(ll<rr)
        {
            mid=(ll+rr+1)/2;
            if(check(mid)) ll=mid;
            else rr=mid-1;
        }
        printf("%d\n",ll);
    }
    return 0;
}