hdu1403 后缀数组入门题

//1403

思路：字符串的任何一个子串都是这个字符串的某个后缀的前缀，则求A和B的最长公共子串等价于求A的后缀和B的后缀的最长公共前缀的最大值。

做法：将第二个字符串写在第一个字符串后面，中间用一个没有出现过的字符隔开，再求这个新的字符串的后缀数组。

#include "bits/stdc++.h"
using namespace std;
const int maxn = 1e6 + 10;
char str[maxn];
int int_str[maxn];
int len;
int str_rank[maxn], s_r[maxn], rank_str[maxn], a[maxn], b[maxn], h_rank[maxn];

int Count[maxn];
void radix(int *str, int *s_r, int *str_rank, int len, int m)
{
    int i;
    memset(Count, 0, sizeof(Count));
    for(i = 0; i < len; ++i)
        ++Count[str[s_r[i]]];
    for(i = 1; i <= m; ++i)
        Count[i] += Count[i - 1];
    for(i = len - 1; i >= 0; --i) {
        str_rank[--Count[str[s_r[i]]]] = s_r[i];
    }
}

void suffix_array(int *str, int *str_rank, int len, int m)
{
    int i;
    for(i = 0; i < len; ++i)
        s_r[i] = i;
    radix(str, s_r, str_rank, len, m);
    rank_str[str_rank[0]] = 0;
    for(i = 1; i < len; ++i) {
        rank_str[str_rank[i]] = rank_str[str_rank[i - 1]] + (str[str_rank[i - 1]] != str[str_rank[i]]);
    }
    int j;
    for(i = 0; (1 << i) < len; ++i) {
        for(j = 0; j < len; ++j) {
            a[j] = rank_str[j] + 1;
            b[j] = ((1 << i) + j < len)? rank_str[(1 << i) + j] + 1: 0;
            str_rank[j] = j;
        }
        radix(b, str_rank, s_r, len, len);
        radix(a, s_r, str_rank, len, len);
        rank_str[str_rank[0]] = 0;
        for(j = 1; j < len; ++j) {
            rank_str[str_rank[j]] = rank_str[str_rank[j - 1]] + (a[str_rank[j - 1]] != a[str_rank[j]] || b[str_rank[j - 1]] != b[str_rank[j]]);
        }
    }
}

void cal_h(int *str, int *str_rank, int *h_rank, int len)
{
    int i, k = 0;
    for(i = 0; i < len; ++i) {
        rank_str[str_rank[i]] = i;
    }
    for(i = 0; i < len; ++i) {
        k = (k == 0? 0: k - 1);
        if(rank_str[i] != 0) {
            while(str[i + k] == str[str_rank[rank_str[i] - 1] + k])
                ++k;
        }
        h_rank[rank_str[i]] = k;
    }
}


int main()
{
    while(scanf("%s", str) != EOF) {
        len = strlen(str);
        int len1 = len;
        str[len] = 'a' - 1;
        scanf("%s", str + len + 1);
        len = strlen(str);
        int i;
        for(i = 0; i < len; ++i)
            int_str[i] = str[i];
        suffix_array(int_str, str_rank, len, 200);
        cal_h(int_str, str_rank, h_rank, len);

//        for(i = 0; i < len; ++i) {
//            printf("%d: %s %d
", i + 1, str + str_rank[i], h_rank[i]);
//        }

        int res = 0;
        for(i = 1; i < len; ++i) {
            if(res < h_rank[i] && ((str_rank[i] < len1 && str_rank[i - 1] > len1) || (str_rank[i] > len1 && str_rank[i - 1] < len1)))
                res = h_rank[i];
        }
        printf("%d
", res);
    }

}