1031. 两个非重叠子数组的最大和
力扣题目链接
给你一个整数数组 nums 和两个整数 firstLen 和 secondLen,请你找出并返回两个非重叠 子数组 中元素的最大和,长度分别为 firstLen 和 secondLen 。
长度为 firstLen 的子数组可以出现在长为 secondLen 的子数组之前或之后,但二者必须是不重叠的。
子数组是数组的一个 连续 部分。
示例 1:
1 2 3
| 输入:nums = [0,6,5,2,2,5,1,9,4], firstLen = 1, secondLen = 2 输出:20 解释:子数组的一种选择中,[9] 长度为 1,[6,5] 长度为 2。
|
示例 2:
1 2 3
| 输入:nums = [3,8,1,3,2,1,8,9,0], firstLen = 3, secondLen = 2 输出:29 解释:子数组的一种选择中,[3,8,1] 长度为 3,[8,9] 长度为 2。
|
示例 3:
1 2 3
| 输入:nums = [2,1,5,6,0,9,5,0,3,8], firstLen = 4, secondLen = 3 输出:31 解释:子数组的一种选择中,[5,6,0,9] 长度为 4,[0,3,8] 长度为 3。
|
提示:
- 1 <= firstLen, secondLen <= 1000
- 2 <= firstLen + secondLen <= 1000
- firstLen + secondLen <= nums.length <= 1000
- 0 <= nums[i] <= 1000
思路:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
|
var maxSumTwoNoOverlap = function(nums, firstLen, secondLen) { let res = 0, n = nums.length let arr = Array(n+1).fill(0) for(let i = 0; i<n;i++){ arr[i+1] = arr[i] + nums[i] } for(let i = firstLen, t = 0; i + secondLen - 1 < n; i++){ t = Math.max(t,arr[i] - arr[i-firstLen]) res = Math.max(res, t + arr[i+secondLen] - arr[i]) } for(let i = secondLen, t = 0; i + firstLen - 1 < n; i++){ t = Math.max(t,arr[i] - arr[i-secondLen]) res = Math.max(res, t + arr[i+firstLen] - arr[i]) } return res };
|