LeetCode 1031: Maximum sum of two non-overlapping subarrays

原题目

Given an array A of non-negative integers, return the maximum sum of elements in two non-overlapping (contiguous) subarrays, which have lengths L and M. (For clarification, the L-length subarray could occur before or after the M-length subarray.)

Formally, return the largest V for which V = (A[i] + A[i+1] + ... + A[i+L-1]) + (A[j] + A[j+1] + ... + A[j+M-1]) and either:

  • 0 <= i < i + L - 1 < j < j + M - 1 < A.length, or
  • 0 <= j < j + M - 1 < i < i + L - 1 < A.length.

Example 1:

Input: A = [0,6,5,2,2,5,1,9,4], L = 1, M = 2
Output: 20
Explanation: One choice of subarrays is [9] with length 1, and [6,5] with length 2.

Example 2:

Input: A = [3,8,1,3,2,1,8,9,0], L = 3, M = 2
Output: 29
Explanation: One choice of subarrays is [3,8,1] with length 3, and [8,9] with length 2.

Example 3:

Input: A = [2,1,5,6,0,9,5,0,3,8], L = 4, M = 3
Output: 31
Explanation: One choice of subarrays is [5,6,0,9] with length 4, and [3,8] with length 3.

题目翻译

给出一个正整数数组 A,并给出两个数值 L、M;
求这个数组中长度为 L 和 M 的,且无重复元素的子数组的元素之和的最大值。

也就是求 V = (A[i] + A[i+1] + ... + A[i+L-1]) + (A[j] + A[j+1] + ... + A[j+M-1]) 的最大值,其中

  • (A[i] + A[i+1] + ... + A[i+L-1]) 是长度为 L 的子数组
  • (A[j] + A[j+1] + ... + A[j+M-1]) 是长度为 M 的子数组

并且 i、j、L、M 满足:

  • 0 <= i < i + L - 1 < j < j + M - 1 < A.length, or
  • 0 <= j < j + M - 1 < i < i + L - 1 < A.length.

答案 (JavaScript)

/**
 * @param {number[]} A
 * @param {number} L
 * @param {number} M
 * @return {number}
 */
var maxSumTwoNoOverlap = function(A, L, M) {
  const prefixSum = [], length = A.length
  prefixSum[0] = A[0]
  for (let i=1; i {
  const length = A.length
  const leftMax = new Array(length), rightMax = new Array(length);
  for (let i = L-1; i= 0; i--) {
    const tmpSum = prefixSum[i + M - 1] - prefixSum[i] + A[i];
    if (i === length - M) {
      rightMax[i] = tmpSum;
    } else {
      rightMax[i] = Math.max(rightMax[i + 1], tmpSum);
    }
  }
  
  let sum = -Infinity
  for (let i = L - 1; i < length - M; i++) {
    sum = Math.max(sum, leftMax[i] + rightMax[i + 1]);
  }
  return sum
}

代码解析

如何高效的求子数组的和

这道题需要找到子数组元素和的最大值,那么肯定就需要求很多很多很多不同子数组的元素和,然后进行比对。

每次都将选出来的子数组元素逐一相加肯定是不合算的。

这里的方法是,建立一个 prefixSum 数组:

const prefixSum = [];
prefixSum[0] = A[0];
for (let i=1; i

这样,对于数组 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],如果我们想求 2-7 这个子数组的和,只需要用 prefixSum[6] - prefixSum[0] 就行了。

如何避免子数组重叠

这道题最容易想到的方法其实就是 —— 暴力搜索:

即,将所有长度为 L 和 M 的子数组都找出来,两两配对,每一对都判断一下它们是不是重合,如果重合就舍弃;如果没有重合部分,就计算这两个数组的和,然后找出最大值即可。

但是,效率真的低啊。。。虽然打着开心的 Accepted,但是。。。

Runtime: 
100 ms, faster than 16.16% of JavaScript online submissions 
for Maximum Sum of Two Non-Overlapping Subarrays.

不爽 =。=

所以该怎么办呢?

不用暴力搜索,那么就要聪明的搜索。
这道题有个很关键的可以「聪明搜索」的点,就是 non-overlapping:既然必须是无重合的数组对,我们一开始只需要寻找无重合的配对来求和就可以了。

如何寻找?

...怎么才能让同桌的领地和自己的领地不要重合呢?...画个三八线呗 =。=

同理,在数组中找一条「分界线」,在这条线两边分别寻找子数组,并让每个子数组的求和尽可能大。
最后找到所有分界情况下的子数组求和的最大值,就是答案了。

例如:有一个数组,[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],选取数字 4,5 之间作为分界线。那么两个子数组 C、D 分别在 [1, 2, 3, 4][5, 6, 7, 8, 9, 10] 中寻找,就一定能保证数组 C、D 不重合。同理,我们还可以把分界线选在 6,7 之间...

这样,比暴力搜索就节省了很多次无用的比较。

函数 split_and_find_max_sum 其实就是在做画分割线 & 寻找最大可能的和这件事情。

不过在函数中,它先分别从前往后(从后往前)计算了所有可能分界线的左边(右边)可能子元素的最大值。

这个是从前往后找:

  for (let i = L-1; i

这个 leftMax 就可以认为是,当分界线选了 i,i+1 之间的时候,分界线左边所有长度为 L 的子数组元素求和的最大值

这个是从后往前找:

  for (let i = length - M; i >= 0; i--) {
      const tmpSum = prefixSum[i + M - 1] - prefixSum[i] + A[i];
      if (i === length - M) {
        rightMax[i] = tmpSum;
      } else {
        rightMax[i] = Math.max(rightMax[i + 1], tmpSum);
      }
  }

rightMax 和 leftMax 基本同理。只不过 rightMax 选子数组的范围是线右边,同时数组长度是 M。

最后一轮 for 循环才是在比对所有分界线情况,并找出最大值。

  let sum = -Infinity
  for (let i = L - 1; i < length - M; i++) {
    sum = Math.max(sum, leftMax[i] + rightMax[i + 1]);
  }
  return sum

函数 split_and_find_max_sum 需要调用两次,原因是两个子数组长度可能不同,谁在线左边,谁在线右边,需要分情况讨论。

你可能感兴趣的:(LeetCode 1031: Maximum sum of two non-overlapping subarrays)