LeetCode 718. Maximum Length of Repeated Subarray

如果把数组换成字符串,这道题就是 Longest Common Substring 的问题。当然,这个和 最长子序列 Longest Common Subsequence 是不一样的。

类似求极值的题,都要用DP想一想。本题就可以用DP来做。

dp[i][j] 表示a数组0~i 和 b数组前0~j,所能得到的最大相同子数组(末尾能匹配的最大长度)。

如果当前正在遍历的 a[i] 和 b[j] 相等,dp[i][j] = dp[i-1][j-1] + 1,如果不相等,dp[i][j]=0。

下面代码写的时候用的是前i个元素 前j个元素,这样就不用单独初始化了。注意下标即可。

class Solution {
public:
    int findLength(vector<int>& A, vector<int>& B) {
        // dp[i][j] A前i个,B前j个 最长重复子数组
        int maxlen=0;
        vector<vector<int>> dp(A.size()+1,vector<int>(B.size()+1,0));
        for (int i=1;i<=A.size();++i){
            for (int j=1;j<=B.size();++j){
                if (A[i-1]==B[j-1]){
                    dp[i][j] = dp[i-1][j-1]+1;
                    maxlen = max(maxlen,dp[i][j]);
                }else dp[i][j]=0;
            }
        }
        /*
        for (int i=1;i<=A.size();++i){
            for (int j=1;j<=B.size();++j){
                cout << dp[i][j] << ' ';
            }
            cout << endl;
        }
        */
        return maxlen;
    }
};

猜你喜欢

转载自www.cnblogs.com/hankunyan/p/9624081.html