718. 最长重复子数组

718. 最长重复子数组

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

思路

动态规划

可以用dp[i][j]表示A数组从i位置，B数组从j位置开始的最长子数组长度，由此可知，当A[i] == B[j]时，dp[i][j]取决与dp[i+1][j+1]的结果，也即使dp[i][j] = dp[i+1][j+1]+1，当A[i] != B[j]时，dp[i][j] = 0。

代码

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

滑动窗口

可以分别固定A数组让B数组滑动，固定B数组让A数组滑动，然后计算相对位置的重复元素即可。

代码

class Solution {
public:
    int findMax(vector<int>& A, vector<int>& B, int index_A, int index_B, int len) {
        int len_return = 0;
        int k = 0;
        for(int i = 0; i < len; i++) {
            if(A[index_A + i] == B[index_B + i]) {
                k++;
            }
            else {
                k = 0;
            }
            len_return = max(k, len_return);
        }
        return len_return;
    }
    int findLength(vector<int>& A, vector<int>& B) {
        int m = A.size(), n = B.size();
        int ans = 0;
        for(int i = 0; i < m; i++) {
            int len = min(m - i, n);
            int tmp = findMax(A, B, i, 0, len);
            ans = max(ans, tmp);
        }

        for(int j = 0; j < n; j++) {
            int len = min(m, n - j);
            int tmp = findMax(A, B, 0, j, len);
            ans = max(ans, tmp);
        }

        return ans;
    }
};