poj 1226 出现或反转后出现在每个字符串中的最长子串（后缀数组/kmp）

题目：http://poj.org/problem?id=1226
题意：

给定 n 个字符串，求出现或反转后出现在每个字符串中的最长子串。

分析:

这题不同的地方在于要判断是否在反转后的字符串中出现。其实这并没有加大题目的难度。只需要先将每个字符串都反过来写一遍，中间用一个互不相同的且没有出现在字符串中的字符隔开，再将 n 个字符串全部连起来，中间也是用一个互不相同的且没有出现在字符串中的字符隔开，求后缀数组。然后二分答案，再将后缀分组。判断的时候，要看是否有一组后缀在每个原来的字符串或反转后的字符串中出现。这个做法的时间复杂度为 O(nlogn)。
当然这题数据比较小，KMP暴力也可以做，枚举第一个字符串的子串（反串），然后看看是否在所有其它字符串中是否出现过即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;
const int INF = 1e9 + 9;
const int N = 3e4 + 9;

/********************倍增算法*后缀数组模板*******************************/

int sa[N], t1[N], t2[N], c[N], rk[N], height[N];
void build_sa (int s[], int n, int m) {
    int i, k, p, *x = t1, *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 (k = 1; k <= n; k <<= 1) {
        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);
        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 s[], int n) {
    int i, j, k = 0;
    for (i = 0; i <= n; i++) rk[sa[i]] = i;
    for (i = 0; i < n; i++) {
        if (k) k--;
        j = sa[rk[i] - 1];
        while (s[i + k] == s[j + k]) k++;
        height[rk[i]] = k;
    }
}
/********************************************************************************/

const int MAX = 111;
int s[N], n, m, length[MAX * 2]; //正反串保存的时候相邻，这样id/2就表示同一个串了
bool flag[MAX];
char str[N];
int check (int x) { //判断在哪个字符串中
    for (int i = 0; i < 2 * m; i++)
        if (x < length[i]) return i;
}
bool judge() {
    for (int i = 0; i < m; i++) if (!flag[i]) return 0;
    return 1;
}
bool ok (int k) {
    memset (flag, 0, sizeof (flag) );
    for (int i = 1; i <= n; i++) {
        if (height[i] < k) {
            if (judge() ) return 1; //判断是否在所有字符串中都出现过
            memset (flag, 0, sizeof (flag) );
            int id = check (sa[i]);
            flag[id / 2] = 1;
            continue;
        }
        int id = check (sa[i]);
        flag[id / 2] = 1;
    }
    return judge();
}
int main() {
    //freopen ("f.txt", "r", stdin);
    int T;
    scanf ("%d", &T);
    while (T--) {
        int XX = 128;
        n = 0;
        int cnt = 0;
        scanf ("%d", &m);
        for (int i = 0; i < m; i++) {
            scanf ("%s", str);
            int len = strlen (str);
            for (int j = 0; j < len; j++) s[n++] = str[j];
            s[n++] = XX++; //每个字符串后面插入一个不同的符号
            length[cnt++] = n;
            for (int j = 0; j < len; j++) s[n++] = str[len - 1 - j];
            s[n++] = XX++; //每个字符串后面插入一个不同的符号
            length[cnt++] = n;
        }
        if(m==1){ //只有一个字符串
            printf("%d\n",strlen(str));
            continue;
        }
        s[n] = 0;
        build_sa (s, n + 1, XX);
        getHeight (s, n);

        int l = 0, r = n;
        while (l < r) { //二分长度
            int mid = l + (r - l + 1) / 2;
            if (ok (mid) ) l = mid;
            else r = mid - 1;
        }
        printf ("%d\n", l);
    }
    return 0;
}
/*
Sample Input

2
3
ABCD
BCDFF
BRCD
2
rose
orchid

Sample Output

2
2



*/