题目描述
给你一个数组 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^3nums.length == n1 <= nums[i] <= 1001 <= left <= right <= n * (n + 1) / 2
解题思路
暴力解也是最简单直接的思路,采用双层循环获取所有的子数组和值插入一个和值数组。
对和值数组进行升序排序。
获取区间中的子数组和值之和,并对 10^9 + 7 取模后返回即可。
代码实现
var rangeSum = function(nums, n, left, right) {
let arr = [];
// 去找子数组区间和
for(let i = 0; i < nums.length; i++){
arr.push(nums[i]);
let tmp = nums[i];
for(let j = i+1; j< nums.length; j++){
tmp = tmp + nums[j];
arr.push(tmp);
}
}
// 排序
let sArray = arr.sort((a,b)=> a-b);
let sum = 0;
for(let i = left; i <= right; i++){
sum = (sum + sArray[i-1]) % 1000000007;
}
return sum;
};
至此我们就完成了leetcode-1508-子数组和排序后的区间和