D - Life Forms POJ - 3294

https://cn.vjudge.net/contest/311974#problem/D 

坑：间隔符要不同，超出ascii不能用char处理了！！！！ 

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>
#include <map>
#include <set>
#include <stack>
using namespace std;

#define sfi(x) scanf("%d",&x)
#define sfc(x) scanf("%c",x)
#define sfl(x) scanf("%lld",&x)
#define sfs(x) scanf("%s",x)

#define rint register int
#define pb push_back
#define fl() printf("flag!\n")
#define INF 0x3f3f3f3f
#define ll long long
#define mem(x,y) memset(x,y,sizeof(x))
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)

const int maxn=1e6+9;
const int mod=1e9+7;

int s[maxn];
int b[maxn];
int t1[maxn],t2[maxn],rk[maxn],sa[maxn],c[maxn],height[maxn];
int n,m;

void get_sa(int *s,int n,int m)
{
    int *x=t1,*y=t2;
    int i,k;
    for(i=0;i<m;i++) c[i]=0;
    for(i=0;i<n;i++) c[x[i]=s[i]]++;
    for(i=1;i<m;i++) c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(k=1;k<=n;k<<=1)
    {
        int p=0;
        for(i=n-k;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(i=0;i<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[y[i]]]++;
        for(i=1;i<m;i++) c[i]+=c[i-1];
        for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        x[sa[0]]=0;
        p=1;
        for(i=1;i<n;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) ? p-1:p++;
        if(p>=n) break;
        m=p;
    }
}


void get_height(int *s,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rk[sa[i]]=i;
    for(i=0;i<n;i++)
    {
        if(k) --k;
        else k=0;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rk[i]]=k;
    }
}

bool vis[109];
int index[109];
int index_size;
int res[maxn];
int cnt=0;
bool ch(int mid,int n,int k)
{
    mem(vis,0);
    cnt=0;
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        if(height[i]>=mid)
        {
            for(int j=1;j<=index_size;j++)
            {
                if(sa[i]>index[j-1]&&sa[i]<index[j])
                {
                    if(!vis[j]) ans++;
                    vis[j]=1;
                }
                if(sa[i-1]>index[j-1]&&sa[i-1]<index[j])
                {
                    if(!vis[j]) ans++;
                    vis[j]=1;
                }
            }
        }
        else
        {
            if(ans>k/2)
            {
                res[++cnt]=sa[i-1];
            }
            ans=0;
            mem(vis,0);
        }
    }
    if(ans>k/2) res[++cnt]=sa[n];

    if(cnt)
    {
        res[0]=cnt;
        return 1;
    }
    return 0;

}

char tmp[maxn];
int main()
{
    //FAST_IO;
    //freopen("input.txt","r",stdin);

    index[0]=-1;
    int cas=0;
    int k=0;
    while(~sfi(k)&&k)
    {
        index_size=0;
        res[0]=0;
        n=0;
        for(int i=0;i<k;i++)
        {
            scanf("%s",tmp+n);
            for(;tmp[n]!='\0';n++) s[n]=tmp[n];
            s[n]='#'+i;
            index[++index_size]=n;
            n++;
        }

        //cout<<s<<endl;

        // cout<<n<<endl;

        m=400;
        s[n-1]=0;

        get_sa(s,n,m);
        get_height(s,n-1);
        int l=1,r=n;
        int ans=0;
        while(l<=r)
        {
            int mid=(l+r)>>1;
            if(ch(mid,n,k)) ans=mid,l=mid+1;
            else r=mid-1;
        }

        //printf("%d\n",ans);

        if(cas++)
            printf("\n");

        if(!ans) puts("?");
        else
        {
            for(int i=1;i<=res[0];i++)
            {
                for(int j=res[i];j<res[i]+ans;j++)
                {
                    printf("%c",s[j]);
                }
                printf("\n");
            }
        }
    }

    return 0;

}