原题链接:
力扣
描述:
给你一个整数数组 nums 和一个整数 k ,找出三个长度为 k 、互不重叠、且全部数字和(3 * k 项)最大的子数组,并返回这三个子数组。
以下标的数组形式返回结果,数组中的每一项分别指示每个子数组的起始位置(下标从 0 开始)。如果有多个结果,返回字典序最小的一个。
示例 1:
输入:nums = [1,2,1,2,6,7,5,1], k = 2
输出:[0,3,5]
解释:子数组 [1, 2], [2, 6], [7, 5] 对应的起始下标为 [0, 3, 5]。
也可以取 [2, 1], 但是结果 [1, 3, 5] 在字典序上更大。
示例 2:
输入:nums = [1,2,1,2,1,2,1,2,1], k = 2
输出:[0,2,4]
提示:
1 <= nums.length <= 2 * 104
1 <= nums[i] < 216
1 <= k <= floor(nums.length / 3)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-sum-of-3-non-overlapping-subarrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
* 解题思路: * 这次我写的思路超时了,参考官方的例子来写的。设计到的主要是贪心算法。 * 从0开始遍历数组,则读到的值应该是i,i+k,i+2k,这样的读到的数值一定是非嵌套的。 * 我们记录几个值: * sum1:当前区间1的和 * sum2:当前区间2的和 * sum3:当前区间3的和 * max1Sum:当前区间之前的sum1的最大值 * max12Sum:当前区间之前的sum1+sum2的最大值 * max123Sum:当前区间之前的sum1+sum2+sum3的最大值 * max1Index:max1Sum改变时,记录其位置 * max22Index:max12Sum改变时,记录其第二个区间的位置 * max21Index:max12Sum改变时,记录其第一个区间的位置 * result:max12Sum改变时,记录其1,2,3个区间的位置 * 当i=k-1的时候,这时候我们可以得到第一个sum的值,即0到k-1之间的和为sum1,k到2k-1之间额和为sum2,2k到3k-1之间额和为sum3。同时记录max12Sum=sum1+sum2,max123Sum=max12Sum+sum3. * 继续往后遍历的时候,就要用到贪心算法了,即每次遍历的时候,我们先判断sum1是否更大,如果时,则替换max1Sum。同时判断此时的max1Sum+sum2是否大于max12Sum,如果大于则替换max12Sum。 * 这时候我们有个疑问,因为sum2是取的当前值,那么之前会不会有更大的?我们一开始的假设就是max12Sum是当前区间之前最大的,读取一个新值后,那么只存在max1Sum+sum2>max12Sum这一种情况,即sum1的最大值+sum2有可能大于max12Sum,否则之前的max12Sum就不是最大的。 * 同理,max123Sum也是一个原理。
代码:
public int[] maxSumOfThreeSubarrays(int[] nums, int k) {
int sum1 = 0;
int sum2 = 0;
int sum3 = 0;
int max1Sum = 0;
int max12Sum = 0;
int max123Sum = 0;
int max1Index = 0;
int max22Index = 0;
int max21Index = 0;
int[] result = new int[3];
for (int i = 0; i < nums.length - 2 * k; i++) {
int currentValue1 = nums[i];
int currentValue2 = nums[i + k];
int currentValue3 = nums[i + k * 2];
sum1 += currentValue1;
sum2 += currentValue2;
sum3 += currentValue3;
if (i >= (k - 1)) {
if (sum1 > max1Sum) {
max1Sum = sum1;
max1Index = i + 1 - k;
}
if (max1Sum + sum2 > max12Sum) {
max22Index = i + 1;
max21Index = max1Index;
max12Sum = max1Sum + sum2;
}
if (max12Sum + sum3 > max123Sum) {
result[0] = max21Index;
result[1] = max22Index;
result[2] = i + 1 + k;
max123Sum = max12Sum + sum3;
}
sum1 -= nums[i + 1 - k];
sum2 -= nums[i + 1];
sum3 -= nums[i + 1 + k];
}
}
return result;
}