[路飞][LeetCode]1508_子数组和排序后的区间和

126 阅读1分钟

「这是我参与2022首次更文挑战的第36天,活动详情查看: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

提示:

  • 1 <= nums.length <= 10^3
  • nums.length == n
  • 1 <= nums[i] <= 100
  • 1 <= left <= right <= n * (n + 1) / 2

解题思路

  • 我们先找所有的子数组和
  • 然后对其进行排序
  • leftright之间的数字进行求和
  • 求和的结果对10^9 + 7取模后返回

解题代码

var rangeSum = function(nums, n, left, right) {
    let arr = []
    // 找所有的子数组和
    for(let i = 0; i<n;i++){
        let num = nums[i]
        arr.push(num)
        for(let j = i+1;j<n;j++){
            num += nums[j]
            arr.push(num)
        }
    }
    arr.sort((a,b)=>a-b) // 排序
    let res = 0
    for(let i = left-1;i<right;i++){
        res += arr[i]
    }
    return res%(Math.pow(10,9)+7)
};

我还成千上万的花没见过,不知道爱的谜底藏在哪个角落,此刻,先为自己而活

如有任何问题或建议,欢迎留言讨论!