UVa11107 后缀数组

这题由于这周六考大物，整整拖了一周时间，一开始写的我估计也没什么大问题，细节比较多：

1.字符数组直接写成int数组，并且最后一位s[n]=0，来维护sa数组，height数组（记录LCS），rank数组；

2.二分的技巧，也看到了很多；

3.看来刘汝佳的书写的也不是很完美，对于后缀数组虽然给了标程，但是具体怎么使用，要注意哪些细节并没有给出提示，好在还有一篇国家队论文可以看，论文里讲的就比较详细。

恩！第一道后缀数组。

#include <cstdio>
#include <fstream>
#include <algorithm>
#include <cstring>
using namespace std;

#define N 200
#define maxn 200000

//correct
int s[maxn];
int sa[maxn], source[maxn], flag[N];
int Rank[maxn],height[maxn];
int t[maxn], t2[maxn], c[maxn];
int n, js;


void build_sa(int n,int m)
{
    int i,*x=t, *y=t2;

    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 (int 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=0;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);
        p=1;x[sa[0]]=0;
        for (i=1;i<n;i++)
            x[sa[i]]= y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1:p++;
        if (p>=n) break;
        m=p;
    }
}


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

void print(int ans)
{
    int cnt=0,k=1;
    memset(flag,0,sizeof(flag));
    for (int i=1;i<n-1;i++)
        if (height[i]>=ans) {
            if (flag[source[sa[i]]]!=k && source[sa[i]]!=source[sa[i-1]]){
                flag[source[sa[i]]]=k;
                if (!cnt) {
                    //correct
                    flag[source[sa[i-1]]]=k;
                    cnt++;
                }
                cnt++;
            }
        }
        else {
                if (cnt>js/2) {
                    for (int j=0;j<ans;j++)
                        printf("%c",s[sa[i-1]+j]+'a'-1);
                    printf("\n");
                }
                k++;
                cnt=0;
        }
}



bool valid(int ans)
{
    int cnt=0,k=1;
    memset(flag,0,sizeof(flag));
    for (int i=1;i<n-1;i++)
        if (height[i]>=ans) {
            //corrcet
            if (flag[source[sa[i]]]!=k && source[sa[i]]!=source[sa[i-1]]){
                flag[source[sa[i]]]=k;
                if (!cnt) {
                    //correct
                    flag[source[sa[i-1]]]=k;
                    cnt++;
                }
                cnt++;
            }
        }
        else {
                if (cnt>js/2) return true;
                k++;
                cnt=0;
        }
    return false;
}

void del()
{
    char temp[1005];
    n=0;
    for (int i=0;i<js;i++){
        scanf("%s",temp);
        int j;
        for (j=0;temp[j]!='\0';j++){
            //correct
            s[n]=temp[j]-'a'+1;
            source[n++]=i;
        }
        s[n]=27+i;
        source[n++]=i;
    }
    s[n]=0;
    build_sa(n+1,27+n);
    getheight();
}


int main()
{
    freopen("1.in","r",stdin);

    int kase=0;
    while (scanf("%d",&js)!=EOF && js){
        if (kase) printf("\n");
        kase++;
        del();

        if (!valid(1)) {
            printf("?\n");
            continue;
        }
        //js
        if (js==1) {
            for (int i=0;i<n-1;i++)
                printf("%c",s[i]+'a'-1);
            printf("\n");
            continue;
        }
        int p=0, q=n, mid, ans=1;
        while (p<=q){
            mid=(p+q)>>1;
            if (valid(mid)) {
                ans = mid;
                p = mid+1;
            }
            else q=mid-1;
        }
        print(ans);
    }

    return 0;
}