题目描述:

给定两个字符串s1和s2,计算其最长公共子串的长度,并返回所有可能的最长公共子串。

# -*- coding: utf-8 -*-
# @Time         : 2019-09-22 22:57
# @Author       : Jayce Wong
# @ProjectName  : job
# @FileName     : longestCommonSubstring.py
# @Blog         : https://sysujayce.github.io/
# @Github       : https://github.com/SysuJayce


def lcs(s1, s2):
    """
    现在我们知道了,如果遇到输入是两个字符串的,需要用到的动态规划的话,那么我们需要的状态是一个
    二维的矩阵。
    首先我们需要定义这个矩阵中每个元素的意义:
    dp[i][j]代表了s1[: i + 1]和s2[: j + 1]以s1[i]和s2[j]结尾的公共子串的长度。
    那么关键就在于如何确定转换方程和如何初始化这个状态矩阵了。

    显然,由于dp[i][j]计算的是同时以s1[i]和s2[j]为结尾公共子串的长度,
    如果s1[i] != s2[j],那么dp[i][j] = 0
    当s1[i] == s2[j]时,dp[i][j] = dp[i - 1][j - 1] + 1
    :param s1: 输入的第一个字符串
    :param s2: 输入的第二个字符串
    :return: 最大公共子串长度、以及最大公共子串的具体值
    """
    # 为了方便编程,先在s1和s2前面加入一个空格占位
    s1 = ' ' + s1
    s2 = ' ' + s2
    rows = len(s1)
    cols = len(s2)
    dp = [[0] * cols for _ in range(rows)]
    maxlen = 0
    for i in range(1, rows):
        for j in range(1, cols):
            if s1[i] == s2[j]:
                dp[i][j] = dp[i - 1][j - 1] + 1
                maxlen = max(maxlen, dp[i][j])
            else:
                dp[i][j] = 0

    res = []
    for i in range(1, rows):
        for j in range(1, cols):
            # s1[i]为结尾的子串,截取长度为maxlen即可
            if dp[i][j] == maxlen:
                res.append(s1[i - maxlen + 1: i + 1])

    return maxlen, res


def main():
    s1 = 'ABCBDEFBWD'
    s2 = 'BCBWD'
    maxlen, res = lcs(s1, s2)
    print(maxlen)
    print(res)


if __name__ == '__main__':
    main()