hdu6138（后缀数组）

hdu6138

题意

给出若干个字符串，每次查询两个字符串，求两个字符串的公共子串且在给出的某一个字符串中作为前缀的最大长度。

分析

求公共子串：后缀数组
判断前缀：字典树

求完后缀数组，遍历下 (height) 数组，对于每个公共子串，都去跑下字典树。

这么暴力竟然没超时，还跑得飞快，数据好水。

code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9 + 7;
const int MAXN = 1e5 + 5;
char s[MAXN];
int sa[MAXN], t[MAXN], t2[MAXN], c[MAXN], n; // n 为 字符串长度 + 1，即最后一位为数字 0
int rnk[MAXN], height[MAXN];
// 构造字符串 s 的后缀数组。每个字符值必须为 0 ~ m-1
void build_sa(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, j, k = 0;
    for(i = 0; i < n; i++) rnk[sa[i]] = i;
    for(i = 0; i < n - 1; i++) {
        if(k) k--;
        j = sa[rnk[i] - 1];
        while(s[i + k] == s[j + k]) k++;
        height[rnk[i]] = k;
    }
}

struct Trie {
    int nxt[MAXN][26], val[MAXN];
    int root, L;
    int newnode() {
        memset(nxt[L], -1, sizeof nxt[L]);
        return L++;
    }
    void init() {
        L = 0;
        root = newnode();
        memset(val, 0, sizeof val);
    }
    void insert(char S[]) {
        int len = strlen(S);
        int now = root;
        for(int i = 0; i < len; i++) {
            int d = S[i] - 'a';
            if(nxt[now][d] == -1) nxt[now][d] = newnode();
            now = nxt[now][d];
            val[now]++;
        }
    }
    int query(int l, int r) {
        int now = root;
        int res = 0;
        for(int i = l; i <= r; i++) {
            int d = s[i] - 'a';
            if(nxt[now][d] == -1) return i - l;
            now = nxt[now][d];
        }
        return r - l + 1;
    }
}trie;
vector<string> v;
int main() {
    int T;
    scanf("%d", &T);
    while(T--) {
        trie.init();
        v.clear();
        int nn;
        scanf("%d", &nn);
        for(int i = 0; i < nn; i++) {
            scanf("%s", s);
            trie.insert(s);
            v.push_back(s);
        }
        int m;
        scanf("%d", &m);
        while(m--) {
            int x, y;
            scanf("%d%d", &x, &y); x--; y--;
            int l = v[x].length();
            for(int i = 0; i < l; i++) {
                s[i] = v[x][i];
            }
            s[l++] = '$';
            for(int i = l; i < l + v[y].length(); i++) {
                s[i] = v[y][i - l];
            }
            int l1 = l;
            l += v[y].length();
            s[l] = 0;
            n = l + 1;
            build_sa(128);
            getHeight();
            int res = 0;
            for(int i = 2; i < n; i++) {
                if(height[i] > 0 && ((sa[i] < l1) != (sa[i - 1] < l1))) {
                    res = max(res, trie.query(sa[i], sa[i] + height[i] - 1));
                }
            }
            printf("%d
", res);
        }
    }
    return 0;
}