BOJ)5582 공통 부분 문자열

문제: icpc.me/5582

두 문자열이 주어질 때 공통 부분 문자열의 최대 길이를 출력하는 문제이다.

두 문자열 사이에 나올 수 없는 한 문자열을 더미로 끼운 뒤 두 문자열을 연결하여 (ex ABC!BCD)

Suffix Array를 이용하여 LCP Array를 구할 때 비교 되는 접미사 SA[i]와 SA[i-1]가 서로 다른 문자열로부터 파생된 것들 중 LCP의 최대길이를 출력하면 된다.

#include <cstdio>
#include <algorithm>
#include <cstring>
#define MAX_N 10000
using namespace std;
char str[2 * MAX_N], str2[MAX_N], res[MAX_N];
int t, n, m, g[MAX_N], tg[MAX_N], SA[MAX_N], r[MAX_N], LCP[MAX_N];
bool cmp(int x, int y) {
    if (g[x] == g[y]) {
        return g[x + t] < g[y + t];
    }
    return g[x] < g[y];
}
int main() {
    t = 1;
    scanf("%s%s", &str, &str2);
    m = strlen(str);
    str[m] = 'Z' + 1;
    strcat(str, str2);
    n = strlen(str);
    for (int i = 0; i < n; i++) {
        SA[i] = i;
        g[i] = str[i] - 'A';
    }        
    while (t <= n) {    
        g[n] = -1;
        sort(SA, SA + n, cmp);    
        tg[SA[0]] = 0;
        for (int i = 1; i < n; i++) {    
            if (cmp(SA[i - 1], SA[i]))
                tg[SA[i]] = tg[SA[i - 1]] + 1;
            else
                tg[SA[i]] = tg[SA[i - 1]];
        }
        for (int i = 0; i < n; i++)
            g[i] = tg[i];    
        t <<= 1;
    }
    for (int i = 0; i < n; i++)
        r[SA[i]] = i;
    int len = 0;
    for (int i = 0; i < n; i++) {
        int k = r[i];
        if (k) {
            int j = SA[k - 1];
            while (str[j + len] == str[i + len])
                len++;
            LCP[k] = len;
            if (len)
                len--;
        }
    }
    int ans = 0;
    for (int i = 1; i < n; i++) {
        if ((SA[i - 1]>m && SA[i] < m) || (SA[i - 1]<m && SA[i]>m)) {
            ans = max(ans, LCP[i]);
        }
    }
    printf("%d\n", ans);
    return 0;
}