题目

题目来源:力扣(LeetCode)

一个数组的 最小乘积 定义为这个数组中 最小值 乘以 数组的 和 。

比方说,数组 [3,2,5] (最小值是 2)的最小乘积为 2 (3+2+5) = 2 10 = 20 。
给你一个正整数数组 nums ,请你返回 nums 任意 非空子数组 的最小乘积 的 最大值 。由于答案可能很大,请你返回答案对 109 + 7 取余 的结果。

请注意,最小乘积的最大值考虑的是取余操作 之前 的结果。题目保证最小乘积的最大值在 不取余 的情况下可以用 64 位有符号整数 保存。

子数组 定义为一个数组的 连续 部分。

示例 1:

输入:nums = [1,2,3,2]
输出:14
解释:最小乘积的最大值由子数组 [2,3,2] (最小值是 2)得到。
2 (2+3+2) = 2 7 = 14 。

示例 2:

输入:nums = [2,3,3,1,2]
输出:18
解释:最小乘积的最大值由子数组 [3,3] (最小值是 3)得到。
3 (3+3) = 3 6 = 18 。

示例 3:

输入:nums = [3,1,5,6,4,2]
输出:60
解释:最小乘积的最大值由子数组 [5,6,4] (最小值是 4)得到。
4 (5+6+4) = 4 15 = 60 。

思路分析

  1. 整个子数组的最小值会出现在子数组的最左边或最右边
  2. 想找到这样的子数组,其实就是数组中一个值作为最小值时,找到左边或右边第一个严格小于该数的元素的位置
  3. 最小乘积 = 数组的最小值 * 数组所有元素的和值。
  4. 假设当前值 就是 子数组的 最小值,其实就是求 以当前值为最小值的子数组 最长可以切多长。
  5. 向左找到第一个小于当前值的位置,向右找到第一个小于当前值的位置。中间这段数组的和值,就是算最 小乘积的第二个值,再乘上当前值就是结果。
  6. 以每个数字作为子数组的最小值,求一个最大乘积,在所有乘积里面找一个最大值,这就是这道题要求的。
  1. /**
  2. * @param {number[]} nums
  3. * @return {number}
  4. */
  5. var maxSumMinProduct = function (nums) {
  6. const mod = 1e9 + 7;
  7. // 使用前缀和sum计算出i之前的所有元素的和
  8. const sum = [0];
  9. for (let i = 1; i <= nums.length; i++) {
  10. sum[i] = sum[i - 1] + nums[i - 1];
  11. }
  12. // 求当前元素右侧第一个严格小于该数的元素的位置
  13. let stack = [];// 单调递增栈,栈内存放的是下标
  14. const len = nums.length;
  15. // right 是nums[i]右侧第一个小于nums[i]的数字的下标 如没有则默认为 len
  16. const right = new Array(len).fill(len); // 默认为 len,即没找到
  17. // 从左往右遍历,求元素右边第一个比其小的元素下标
  18. for (let i = 0; i < len; i++) {
  19. // 如果当前元素比栈顶元素小,栈顶元素弹出
  20. while (stack.length && nums[stack[stack.length - 1]] > nums[i]) {
  21. // 记录nums[i]的右边界
  22. right[stack[stack.length - 1]] = i;
  23. // 弹栈
  24. stack.pop();
  25. }
  26. stack.push(i)
  27. }
  28. // 求当前元素左侧第一个严格小于该数的元素的位置
  29. stack = []; // 将栈清空,求元素左边第一个比其小的元素下标
  30. // left 是nums[i]右侧第一个小于nums[i]的数字的下标 如没有则默认为 -1
  31. const left = new Array(len).fill(-1); // 默认为-1,即没找到
  32. // 从右往左遍历,求元素左边第一个比其小的元素下标
  33. for (let i = len - 1; i >= 0; i--) {
  34. // 如果当前元素比栈顶元素小,栈顶元素弹出
  35. while (stack.length && nums[stack[stack.length - 1]] > nums[i]) {
  36. // 记录nums[i]的左边界
  37. left[stack[stack.length - 1]] = i;
  38. // 弹栈
  39. stack.pop();
  40. }
  41. stack.push(i);
  42. }
  43. // 根据前缀和 和left, right数组进行枚举求解最大乘积
  44. let max = BigInt(0);
  45. for (let k = 0; k < len; k++) {
  46. // 以当前元素为最小值的最大乘积
  47. // 最小元素 乘以 区间和
  48. const total = BigInt(sum[right[k]] - sum[left[k] + 1]) * BigInt(nums[k]);
  49. if (max < total) {
  50. max = total;
  51. }
  52. }
  53. return max % BigInt(mod);
  54. };