[路飞]_LeetCode_1508. 子数组和排序后的区间和

137 阅读2分钟

「这是我参与2022首次更文挑战的第35天,活动详情查看:2022首次更文挑战

题目

给你一个数组 nums ,它包含 n 个正整数。你需要计算所有非空连续子数组的和,并将它们按升序排序,得到一个新的包含 n * (n + 1) / 2 个数字的数组。

请你返回在新数组中下标为 left 到 right (下标从 1 开始)的所有数字和(包括左右端点)。由于答案可能很大,请你将它对 10^9 + 7 取模后返回。

示例 1:

输入:nums = [1,2,3,4], n = 4, left = 1, right = 5
输出:13 
解释:所有的子数组和为 1, 3, 6, 10, 2, 5, 9, 3, 7, 4 。
将它们升序排序后,我们得到新的数组 [1, 2, 3, 3, 4, 5, 6, 7, 9, 10] 。
下标从 le = 1 到 ri = 5 的和为 1 + 2 + 3 + 3 + 4 = 13

示例 2:

输入:nums = [1,2,3,4], n = 4, left = 3, right = 4
输出:6
解释:给定数组与示例 1 一样,所以新数组为 [1, 2, 3, 3, 4, 5, 6, 7, 9, 10] 。
下标从 le = 3 到 ri = 4 的和为 3 + 3 = 6

示例 3:

输入:nums = [1,2,3,4], n = 4, left = 1, right = 10
输出:50

来源:力扣(LeetCode)leetcode-cn.com/problems/ra…

解题思路

解题步骤如下:

  • 第一步:通过双循环计算子数组的和,将结果放在一个数组内;
  • 第二步:对第一步的求和结果排序;
  • 第三步:对 left 至 right 区间的和再求和,注意 left 从 1 开始,求和时数组的开始位置是 left - 1,结束位置是 right - 1;
  • 第四部:题目要求对结果 mod 一个 1000000007;

代码实现

var rangeSum = function (nums, n, left, right) {
    let len = nums.length
    const arr = []

    //通过双循环计算子数组的和
    for (let i = 0; i < len; i++) {
        let tmp = nums[i]
        arr.push(tmp)

        for (let j = i + 1; j < len; j++) {
            tmp += nums[j]
            arr.push(tmp)
        }
    }

    //子数组和排序
    arr.sort((a, b) => a - b)

    let ans = 0
    // 计算 left 到 right 的和,注意 left 是从 1 开始,所以开始位置是 left - 1
    for (let i = left - 1; i < right; i++) {
        ans += arr[i]
    }

    // 对结果 mod 一个 1000000007
    return ans % 1000000007
};

如有错误欢迎指出,欢迎一起讨论!