718. 最长重复子数组

    技术2022-07-11  115

    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; } };
    Processed: 0.020, SQL: 9