leetcode 718. 最长重复子数组

一、题目要求

给两个整数数组 A 和 B ，返回两个数组中公共的、长度最长的子数组的长度。

示例：

输入： A: [1,2,3,2,1] B: [3,2,1,4,7] 输出：3 解释： 长度最长的公共子数组是 [3, 2, 1] 。

二、代码

class Solution {
public:
    int findLength(vector<int>& A, vector<int>& B) {
        vector<vector<int>> dp(A.size()+1,vector<int>(B.size()+1,0));
        
        int res=0;
        for(int i=0;i<A.size();++i){
            for(int j=0;j<B.size();++j){
                if(A[i]==B[j]){
                    dp[i+1][j+1]=dp[i][j]+1;
                    res=max(res,dp[i+1][j+1]);
                }
            }
        }

        return res;
    }
};

三、解题思路

最长公共子序列长度

hello，ahslo，结果是hlo，答案是3（因为子序列不必是连续的）

res(i,j)表示字符串a的前i个字符和字符串b的前j个字符的结果，最终求解的是res(5,5)

首先列出状态转移方程：

$$res(i,j)=\{\begin{array}{rlrlrlr}0& ,& i=0& & or& & j=0\\ res(i-1,j-1)+1& ,& A_i=B_j& & & & \\ \max (res(i-1,j),res(i,j-1))& ,& A_i\ne B_j& & & & \end{array}$$

	0	h	e	l	l	o
0	0	0	0	0	0	0
a	0	0	0	0	0	0
h	0	1	1	1	1	1
s	0	1	1	1	1	1
l	0	1	1	2	2	2
o	0	1	1	2	2	3

根据dp数组，可以回溯得到最长公共子序列是"hlo"

最长公共子串长度

同样是hello和ahslo，结果是lo，答案是2（子串必须是连续的）

res(i,j)表示字符串A的前i个字符和字符串B的前j个字符中，最后字符是$A_i$（即$B_j$）的最长公共子字符串的长度，最终求解$res[][]$数组中的最大值

状态转移方程：（如果 $A_i\ne B_j$ ，则说明此时子字符串的长度为0）

$$res(i,j)=\{\begin{array}{rlrlrlr}0& ,& i=0& & or& & j=0\\ res(i-1,j-1)+1& ,& A_i=B_j& & & & \\ 0& ,& A_i\ne B_j& & & & \end{array}$$

	0	h	e	l	l	o
0	0	0	0	0	0	0
a	0	0	0	0	0	0
h	0	1	0	0	0	0
s	0	0	0	0	0	0
l	0	0	0	1	1	0
o	0	0	0	0	0	2

可以看到，本题即为最长公共子串问题，用动态规划dp求解即可。