最长公共子串

最长公共子串

题目描述

给定两个字符串str1和str2,输出两个字符串的最长公共子串，如果最长公共子串为空，输出-1。

输入描述：

输入包括两行，第一行代表字符串srr1，第二行代表字符串str2。$\left(1\le length(str1),length(str2)\le 5000\right)$

输出描述：

输出包括一行，代表最长公共子串。

示例1

输入

1AB2345CD
12345EF

输出

2345

备注：

时间复杂度 $O(n^2)$ ，额外空间复杂度$O(1)$。（n可以为其中任意一个字符串长度）

---

题解：

状态转移方程为：
$$F[i,j]=\{\begin{array}{rlr}F[i-1,j-1]+1& & str1[i]=str2[j]\\ 0& & str1[i]\ne str2[j]\end{array}$$
普通解法： 使用额外的二维状态数组进行状态表示（这种写法很简单，在输出时，需要记录最大值以及最大长度的终点）

普通解法代码：

#include <cstdio>
#include <cstring>

using namespace std;

const int N = 5010;

int f[N][N];
char str1[N];
char str2[N];

int main(void) {
    scanf("%s", str1);
    scanf("%s", str2);
    int len1 = strlen(str1);
    int len2 = strlen(str2);
    int max_len = 0, idx = 0;
    for (int i = 1; i <= len1; ++i) {
        for (int j = 1; j <= len2; ++j) {
            if (str1[i - 1] == str2[j - 1]) f[i][j] = f[i - 1][j - 1] + 1;
            if (f[i][j] > max_len) {
                max_len = f[i][j];
                idx = i;
            }
        }
    }
    if (idx == 0) return 0 * puts("-1");
    for (int i = idx - max_len; i < idx; ++i) putchar(str1[i]);
    return 0 * puts("");
}

进阶解法：

题目要求额外的空间复杂度为 $O(1)$ ，上面的解法肯定不行，但是我们观察上面的状态转移方程，发现每个状态只跟左上角的状态有关，我们可以按照斜线方向计算所有的值，于是我们可以只用一个变量就可以计算出所有位置的值。

进阶解法代码：

#include <cstdio>
#include <cstring>

using namespace std;

const int N = 5010;

char str1[N];
char str2[N];

int main(void) {
    scanf("%s", str1);
    scanf("%s", str2);
    int len1 = strlen(str1);
    int len2 = strlen(str2);
    int row = 0, col = len2 - 1;
    int max_len = 0, ends = 0;
    while (row < len1) {
        int i = row, j = col;
        int ans = 0;
        while (i < len1 && j < len2) {
            if (str1[i] != str2[j]) ans = 0;
            else ans += 1;
            if (ans > max_len) {
                max_len = ans;
                ends = i;
            }
            ++i, ++j;
        }
        if (col) --col;
        else ++row;
    }
    if (!ends) return 0 * puts("-1");
    for (int i = ends - max_len + 1; i <= ends; ++i) putchar(str1[i]);
    return 0 * puts("");
}