hdu 3518 Boring counting 后缀数组基础题

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 2549    Accepted Submission(s): 1030

Problem Description

035 now faced a tough problem,his english teacher gives him a string,which consists with n lower case letter,he must figure out how many substrings appear at least twice,moreover,such apearances can not overlap each other.
Take aaaa as an example.”a” apears four times,”aa” apears two times without overlaping.however,aaa can’t apear more than one time without overlaping.since we can get “aaa” from [0-2](The position of string begins with 0) and [1-3]. But the interval [0-2] and [1-3] overlaps each other.So “aaa” can not take into account.Therefore,the answer is 2(“a”,and “aa”).

 

Input

The input data consist with several test cases.The input ends with a line “#”.each test case contain a string consists with lower letter,the length n won’t exceed 1000(n <= 1000).

 

Output

For each test case output an integer ans,which represent the answer for the test case.you’d better use int64 to avoid unnecessary trouble.

 

Sample Input

aaaa

ababcabb

aaaaaa

#

 

Sample Output

2

3

3

 

Source

2010 ACM-ICPC Multi-University Training Contest（9）——Host by HNU

 

思路：熟悉heigh数组就好了。枚举子串的长度k，将heigh数组按顺序分组，若该组的heigh值都大于等于k，则具有长度为k的公共前缀。比如heigh：3,2,3,1,2,0,1  当k=2时，分成(3,2,3) (1) (2) (0,1)

#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
const int x = ;
const int maxn = ;
int t1[maxn], t2[maxn], c[maxn];
bool cmp(int *r, int a, int b, int l) {
    return r[a] == r[b] && r[a + l] == r[b + l];
}
void da(char str[], int sa[], int Rank[], int heigh[], int n, int m)
{
    n++;
    int i, j, p, *x = t1, *y = t2;
    for(i = ; i < m; ++i) c[i] = ;
    for(i = ; i < n; ++i) c[ x[i] = str[i] ]++;
    for(int i = ; i < m; ++i) c[i] += c[i - ];
    for(int i = n - ; i >= ; --i) sa[--c[x[i]]] = i;

    for(int j = ; j <= n; j <<= )
    {
        p = ;
        for(i = n - j; i < n; ++i) y[p++] = i;
        for(i = ; i < n; ++i) if(sa[i] >= j) y[p++] = sa[i] - j;

        for(i = ; i < m; ++i) c[i] = ;
        for(i = ; i < n; ++i) c[x[y[i]]]++;
        for(i = ; i < m; ++i) c[i] += c[i - ];
        for(i = n - ; i >= ; --i) sa[--c[x[y[i]]]] = y[i];
        swap(x, y);
        p = ; x[ sa[] ] = ;
        for(i = ; i < n; ++i)
            x[ sa[i] ] = cmp(y, sa[i - ], sa[i], j) ? p -  : p++;
        if(p >= n) break;
        m = p;
    }
    int k = ;
    n--;
    for(i = ; i <= n; ++i) Rank[ sa[i] ] = i;
    for(i = ; i < n; ++i) {
        if(k) k--;
        j = sa[Rank[i] - ];
        while(str[i + k] == str[j + k]) k++;
        heigh[ Rank[i] ] = k;
    }
}

int Rank[maxn], heigh[maxn], sa[maxn];
char s[maxn];
void out(int n) {
    ///Rank数组的有效范围是0~n-1， 值是1~n
    for(int i = ; i <= n; ++i) printf("%d ", Rank[i]);
    puts("");
    ///sa数组的有效范围是1~n，值是0~n-1
    for(int i = ; i <= n; ++i) printf("%d ", sa[i]);
    puts("");
    ///heigh数组的有效范围是2~n
    for(int i = ; i <= n; ++i) printf("%d ", heigh[i]);
}
int calc(int k, int n) {
    int mi = INF, mx = -INF, res = ;
    for(int i = ; i <= n; ++i) {
        if(heigh[i] >= k) {
            mi = min(mi, min(sa[i - ], sa[i]));
            mx = max(mx, max(sa[i - ], sa[i]));
        }else {
            if(mx - mi >= k) res++;
            mx = -INF, mi = INF;
        }
    }
    if(mi != INF && mx - mi >= k) res++;
    return res;
}
int main()
{
   // freopen("in.txt", "r", stdin);
    //freopen("out2.txt", "w", stdout);
    while(scanf("%s", s)) {
        if(strcmp(s, "#") == ) break;
        int n = strlen(s);
        da(s, sa, Rank, heigh, n, );
       // out(n);
        int ans = ;
        for(int i = ; i <= (n >> ); ++i) ans += calc(i, n);
        printf("%d\n", ans);
    }
    return ;
}

一开始的做法是hash+map，n^2logn会tle？ 不明觉厉，暂时保存一下，忘指点。。。

#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
const int x = ;
const int maxn = ;
ull H[maxn], xp[maxn];
map<ull, pair<int, int> > m;
int len;
char s[maxn];
void init() {
    xp[] = ;
    for(int i = ; i <= ; ++i) xp[i] = xp[i - ] * x;
}
int main() {
   // freopen("in.txt", "r", stdin);
   // freopen("out1.txt", "w", stdout);
    init();
    while(scanf("%s", s)) {
        if(strcmp(s, "#") == ) break;
        len = strlen(s);
        H[len] = ;
        for(int i = len - ; i >= ; --i) H[i] = H[i + ] * x + (s[i]);
        m.clear();
        long long ans = ;
        for(int k = ; k <= len / ; ++k) {
            for(int i = ; i <= len - k; i++) {
                ull hash = H[i] - H[i + k] * xp[k];
                int lp = m[hash].first;
                int is = m[hash].second;
                if(is ==  && i >= lp + k) {
                    m[hash].second = -;
                    ans++;
                }else if(is == -) {
                    continue;
                }else if(is == ) {
                    m[hash].first = i;
                    m[hash].second = ;
                }
            }
        }
        printf("%I64d\n", ans);
    }
    return ;
}

poj1743

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
const int x = ;
const int maxn =  + ;
int t1[maxn], t2[maxn], c[maxn];
bool cmp(int *r, int a, int b, int l) {
    return r[a] == r[b] && r[a + l] == r[b + l];
}
void da(int str[], int sa[], int Rank[], int heigh[], int n, int m)
{
    n++;
    int i, j, p, *x = t1, *y = t2;
    for(i = ; i < m; ++i) c[i] = ;
    for(i = ; i < n; ++i) c[ x[i] = str[i] ]++;
    for(int i = ; i < m; ++i) c[i] += c[i - ];
    for(int i = n - ; i >= ; --i) sa[--c[x[i]]] = i;

    for(int j = ; j <= n; j <<= )
    {
        p = ;
        for(i = n - j; i < n; ++i) y[p++] = i;
        for(i = ; i < n; ++i) if(sa[i] >= j) y[p++] = sa[i] - j;

        for(i = ; i < m; ++i) c[i] = ;
        for(i = ; i < n; ++i) c[x[y[i]]]++;
        for(i = ; i < m; ++i) c[i] += c[i - ];
        for(i = n - ; i >= ; --i) sa[--c[x[y[i]]]] = y[i];
        swap(x, y);
        p = ; x[ sa[] ] = ;
        for(i = ; i < n; ++i)
            x[ sa[i] ] = cmp(y, sa[i - ], sa[i], j) ? p -  : p++;
        if(p >= n) break;
        m = p;
    }
    int k = ;
    n--;
    for(i = ; i <= n; ++i) Rank[ sa[i] ] = i;
    for(i = ; i < n; ++i) {
        if(k) k--;
        j = sa[Rank[i] - ];
        while(str[i + k] == str[j + k]) k++;
        heigh[ Rank[i] ] = k;
    }
}

int Rank[maxn], heigh[maxn], sa[maxn];
int s[maxn];
void out(int n) {
    ///Rank数组的有效范围是0~n-1， 值是1~n
    for(int i = ; i <= n; ++i) printf("%d ", Rank[i]);
    puts("");
    ///sa数组的有效范围是1~n，值是0~n-1
    for(int i = ; i <= n; ++i) printf("%d ", sa[i]);
    puts("");
    ///heigh数组的有效范围是2~n
    for(int i = ; i <= n; ++i) printf("%d ", heigh[i]);
}
int flag, ans;
bool check(int k, int n) {
    flag = ;
    int mi = INF, mx = -INF;
    for(int i = ; i <= n; ++i) {
        if(heigh[i] >= k) {
            mi = min(mi, min(sa[i - ], sa[i]));
            mx = max(mx, max(sa[i - ], sa[i]));
        }else {
            if(mx - mi >= k + ) return true;
            mx = -INF, mi = INF;
        }
    }
    if(mi != INF && mx - mi >= k + ) return true;
    return false;
}
void solve(int n) {
    int L = , R = n + ;
    while(R - L > ) {
        int M = (L + R) >> ;
        if(check(M, n)) L = M;
        else R = M;
    }
    ans = L;
}
int main() {
    int n;
    while(~scanf("%d", &n) && n) {
        for(int i = ; i < n; ++i) scanf("%d", &s[i]);
        for(int i = ; i < n - ; ++i) {
                s[i] = s[i + ] - s[i];
                s[i] += ;
        }
        //for(int i = 0; i < n - 1; i++) printf("%d ", s[i]);
        n--;
        s[n] = ;
        da(s, sa, Rank, heigh, n, );
            solve(n);
            if(ans +  < ) puts("");
            else
            printf("%d\n", ans + );
    }
    return ;
}