【POJ3294】 后缀数组

POJ3294
链接
题意为给你n个字符串,求至少出现在 [ n 2 ] 个串中的最长公共子串,如果有多个全部输出。

本题和上题几乎一模一样,只是要把判断条件由来自n个串改成来自 [ n 2 ] 个串,注意一下存储答案的方式,如果当前二分量存在答案并且第一次访问,那么清空 v e c t o r ,如果不是第一次访问,就在 v e c t o r 里面 p u s h _ b a c k 就好了。
POJ3294代码

#include <iostream>
#include<algorithm>
#include <stdio.h>
#include <string.h>
#include <vector>
using namespace std;
#define maxn 150005
const int INF = 0x3f3f3f3f;
int wa[maxn],wb[maxn],wsf[maxn],wv[maxn],sa[maxn];
int rank[maxn],height[maxn],s[maxn];
char str[105][1005];
int t,lenn[maxn];
int belong[maxn];
int vis[105];
vector<int> anspos;
int cmp(int *r,int a,int b,int k)
{
    return r[a]==r[b]&&r[a+k]==r[b+k];
}
void getsa(int *r,int *sa,int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0; i<m; i++)  wsf[i]=0;
    for(i=0; i<=n; i++)  wsf[x[i]=r[i]]++;
    for(i=1; i<m; i++)  wsf[i]+=wsf[i-1];
    for(i=n; i>=0; i--)  sa[--wsf[x[i]]]=i;
    p=1;
    j=1;
    for(; p<=n; j*=2,m=p)
    {
        for(p=0,i=n+1-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++)  wsf[i]=0;
        for(i=0; i<=n; i++)  wsf[wv[i]]++;
        for(i=1; i<m; i++)  wsf[i]+=wsf[i-1];
        for(i=n; i>=0; i--)  sa[--wsf[wv[i]]]=y[i];
        t=x;
        x=y;
        y=t;
        x[sa[0]]=0;
        for(p=1,i=1; i<=n; i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)? p-1:p++;
    }
}
void getheight(int *r,int n)
{
    int i,j,k=0;
    for(i=1; i<=n; i++)  rank[sa[i]]=i;
    for(i=0; i<n; i++)
    {
        if(k)
            k--;
        else
            k=0;
        j=sa[rank[i]-1];
        while(r[i+k]==r[j+k])
            k++;
        height[rank[i]]=k;
    }
}
int check(int x,int n)
{
    int flag=0;
    for(int i=1;i<=n-1;i++)
    {
        if(height[i]<x) continue;
        int cnt=0;
        for(int j=0;j<=t;j++) vis[j]=0;
        while(height[i]>=x&&i<=n-1)
        {
            if(!vis[belong[sa[i-1]]])
            {
                vis[belong[sa[i-1]]]=1;
                cnt++;
            }
            i++;
        }
        if(!vis[belong[sa[i-1]]])
        {
            vis[belong[sa[i-1]]]=1;
            cnt++;
        }
        if(cnt>t/2)
        {
            if(flag==0)
            {
                flag=1;
                anspos.clear();
                anspos.push_back(sa[i-1]);
            }
            else
            {
                anspos.push_back(sa[i-1]);
            }
        }
    }
    if(flag==0) return false;
    else return true;
}
int main()
{
    int len,n;
    while(scanf("%d",&t)!=EOF)
    {
        if(t==0)
        {
            printf("\n");
            break;
        }
        n=0;
        int pos=30;
        for(int i=0;i<t;i++)
        {
            scanf("%s",str[i]);
            lenn[i]=strlen(str[i]);
            for(int j=0;j<lenn[i];j++)
             {
                 s[n++]=str[i][j]-'a'+1;
                 belong[n-1]=i;
             }
            s[n++]=pos++;
        }
        s[n]=0;
        getsa(s,sa,n,150);
        getheight(s,n);
        anspos.clear();
        int l=1,r=1000,mid;
        while(l<=r)
        {
            mid=(l+r)>>1;
            if(check(mid,n)) l=mid+1;
            else r=mid-1;
        }
        if(r==0) printf("?\n");
        else
        {
            for(int i=0;i<anspos.size();i++)
            {
                for(int j=anspos[i];j<anspos[i]+r;j++)
                    printf("%c",s[j]-1+'a');
                printf("\n");
            }
        }
        printf("\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_38891827/article/details/80345165