F - Repeats SPOJ - REPEATS

這是一篇論文裏面的題目,雖然看了論文裏面的解法,發現對我還是比較難理解,最後思考了好久才明白這題的具體做法。
我們去枚舉重複子串的長度,如果這個子串的重複長度大於等於2,那麼對每個枚舉的長度L來說,我們計算S[0], S[L], S[2*L], S[3*L]……,這些相鄰的兩個後綴,一定會有一個大於L的前綴(也有可能在當前的前面),設這個前綴爲k那麼長爲L的串就會重複k/L+1次,同時要考慮前面有沒有可能有多餘的沒有匹配上的,比如ebbbb對於L=2時,bbb與b的前綴只有1,是因爲沒有考慮到前面的一位,所以要再往前處理下。


#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
#define Max_N (2*50000 + 100)
int n;
int k;
int a[Max_N];
int rank1[Max_N];
int tmp[Max_N];
bool compare_sa(int i, int j)
{
    if(rank1[i] != rank1[j]) return rank1[i] < rank1[j];
    else {
        int ri = i + k <= n ? rank1[i + k] : -1;
        int rj = j + k <= n ? rank1[j + k] : -1;
        return ri < rj;
    }
}

void construct_sa(int buf[], int s, int sa[])
{
    int len = s;
    for (int i = 0; i <= len; i++) {
        sa[i] = i;
        rank1[i] = i < len ? buf[i] : -1;
    }

    for ( k = 1; k <= len; k *= 2) {
        sort(sa, sa + len +1, compare_sa);
        tmp[sa[0]] = 0;
        for (int i = 1; i <= len; i++) {
            tmp[sa[i]] = tmp[sa[i-1]] + (compare_sa(sa[i-1], sa[i]) ? 1 : 0);
        }
        for (int i = 0; i <= len; i++) {
            rank1[i] = tmp[i];
        }
    }
}

void construct_lcp(int buf[], int len, int *sa, int *lcp)
{
    int h = 0;
    lcp[0] = 0;
    for (int i = 0; i < len; i++) {
        int j = sa[rank1[i] - 1];
        if (h > 0) h--;
        for (; j + h < len && i + h < len; h++) {
            if (buf[j+h] != buf[i+h]) break;
        }
        lcp[rank1[i] - 1] = h;
    }
}
int sa[Max_N];
int rev[Max_N];
int lcp[Max_N];
char str[3];
int f[Max_N][20];
void rmq()
{
    for (int i = 1; i <= n; i++)
        f[i][0] = lcp[i];
    f[n][0] = 1000000;
     for (int j=1; (1<<j) <= n ; j++)  
        for (int i=1; i+(1<<j)-1<=n; i++)  
            f[i][j]= min(f[i][j-1],f[i+(1<<j-1)][j-1]);  
    return;  
}

int lcp1(int L, int R)
{
    if (L> R) {
        int ss = R;
        R = L;
        L = ss;
    }
    R--;
    int k1 = 0;
    while((1<<(k1+1)) <= R-L+1) k1++;
    return min(f[L][k1], f[R-(1<<k1)+1][k1]);
}

int main()
{
    int T;
    cin >> T;
    while (T--) {
        scanf("%d", &n);
        for (int i = 0; i < n; i++) {
            scanf("%s", str);
            a[i] = int(str[0]);
        }
        a[n] = 0;
        construct_sa(a, n, sa);
        construct_lcp(a, n, sa, lcp);
        rmq();
        /*int l,r;
        cin >> l >> r;
        cout << lcp1(l, r) << endl;*/
        int max = 0;
        for (int i = 1; i <= n; i++) {
            for (int j = 0; j + i < n; j += i) {
                int k1 = lcp1(rank1[j], rank1[j+i]);
                int r = k1 / i + 1;
                int t = j - (i - k1%i);
                if (t >= 0)
                    if (lcp1(rank1[t], rank1[t+i]) >= i) r++;
                if (r > max) max = r;
            }
        }
        if (max == 0) max = 1;
        printf("%d\n", max);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章