LeetCode Everyday - 子数组范围和

84 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情

不邻接植花

给你一个整数数组 nums 。nums 中,子数组的 范围 是子数组中最大元素和最小元素的差值。

返回 nums 中 所有 子数组范围的 和 。

子数组是数组中一个连续 非空 的元素序列。

示例1:

输入:nums = [1,2,3]
输出:4
解释:nums 的 6 个子数组如下所示:
[1],范围 = 最大 - 最小 = 1 - 1 = 0 
[2],范围 = 2 - 2 = 0
[3],范围 = 3 - 3 = 0
[1,2],范围 = 2 - 1 = 1
[2,3],范围 = 3 - 2 = 1
[1,2,3],范围 = 3 - 1 = 2
所有范围的和是 0 + 0 + 0 + 1 + 1 + 2 = 4

示例2:

输入:nums = [1,3,3]
输出:4
解释:nums 的 6 个子数组如下所示:
[1],范围 = 最大 - 最小 = 1 - 1 = 0
[3],范围 = 3 - 3 = 0
[3],范围 = 3 - 3 = 0
[1,3],范围 = 3 - 1 = 2
[3,3],范围 = 3 - 3 = 0
[1,3,3],范围 = 3 - 1 = 2
所有范围的和是 0 + 0 + 0 + 2 + 0 + 2 = 4

示例3:

输入: nums = [4,-2,-3,4,1]
输出: 59
解释: nums 中所有子数组范围的和是 59

提示:

  • 1 <= nums.length <= 1000
  • -109 <= nums[i] <= 109

解题思路:

根据题意,由于是连续的元素序列,所以可以采用滑块的方式,获取子数组。 子数组长度逐渐 +1,使用 dp 来记录上一次子数组的最大值与最小值,则在任意一个位置,只需要比较上一次的最大值与最小值,就可以得出该位置子数组的最大与最小值,并记录该位置的差值。

由于采用传统的动态规划数组,nxn 的二维数组会内存溢出,所以采用滚动数组,来减少空间复杂度

我的答案:

/**
 * @param {number[]} nums
 * @return {number}
 */
var subArrayRanges = function(nums) {
    let len = nums.length;
    let dp = new Array(len).fill(null).map(() => {
        return []
    })
    let result = 0;
    for(let i = 0; i < len; i++) {
        for(let j = 0; j < (len-i); j++) {
            if(i == 0) {
                dp[j][0] = nums[j];
                dp[j][1] = nums[j];
            } else {
                dp[j][0] = Math.min(dp[j][0], nums[j+i])
                dp[j][1] = Math.max(dp[j][1], nums[j+i])
            }
            result += dp[j][1] - dp[j][0]
        }
    }
    return result;
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´・ᴗ・` )