LeetCode#718 Maximum Length of Repeated Subarray (week9)

week9

题目

Given two integer arrays A and B, return the maximum length of an subarray that appears in both arrays.

Example 1:
Input:
A: [1,2,3,2,1]
B: [3,2,1,4,7]
Output: 3
Explanation:
The repeated subarray with maximum length is [3, 2, 1].
Note:
1 <= len(A), len(B) <= 1000
0 <= A[i], B[i] < 100
原题地址：https://leetcode.com/problems/maximum-length-of-repeated-subarray/description/

解析

题目要求找到两个数组相同的最长子数组并返回其长度。
大致思路是用两重循环，外循环指定第一个数组的位置，内循环指定第二个数组的位置，以这两个位置分别为两个子数组的起始位置，检索直至相对应的某一位元素不同，得到以这两个位置为起始的最长相同子数组长度。
注意到上述做法可以找到正确答案，但是中间却做了很多不必要的工作，考虑数组[1,2,3,4,1]和[2,1,2,3,4]，按照循环会找到[1,2,3,4]和[1,2,3,4]，也会找到[2,3,4]和[2,3,4]，但可以发现后者是前者一部分，因此检索时可直接跳过，跳过的条件是起始位置均在前者（已经找到的最长相同子数组）的中间，以此节省检索时间。

代码

class Solution {
public:
    int findLength(vector<int>& A, vector<int>& B) {
        int max = 0;
        /*记录已知最长相同子数组在两个原数组中的起始位置和结束位置*/
        int A_s = 0, A_e = 0, B_s = 0, B_e = 0;
        for (int i = 0; i < A.size(); ++i) {
            for (int j = 0; j < B.size(); ++j) {
                int k = 0;
                bool breakFlag = false;
                if (!(i > A_s && i <= A_e && j > B_s && j <= B_e)) {
                    while (A[i + k] == B[j + k]) {
                        if ((i + k + 1) < A.size() && (j + k + 1) < B.size()) {
                            ++k;
                        }
                        else {
                            /*如果是因为检索到数组末尾跳出循环，k为当前最长相同子数组长度-1*/
                            breakFlag = true;
                            break;
                        }
                    }
                    int current = breakFlag ? (++k) : k;
                    if (current > max) {
                        max = current;
                        A_s = i;
                        A_e = i + current - 1 ;
                        B_s = j;
                        B_e = j + current - 1;
                    }
                }
            }
        }
        return max;
    }
};