[路飞]_js算法:leetcode 1508-子数组和排序后的区间和

91 阅读1分钟

leetcode 1508. 子数组和排序后的区间和

问题描述: 给你一个数组 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

思路: 获取前缀和,前缀和数组排序,求范围值

/**
 * @param {number[]} nums
 * @param {number} n
 * @param {number} left
 * @param {number} right
 * @return {number}
 */
var rangeSum = function(nums, n, left, right) {
   let arr=[];
   for(let i=0;i<nums.length;i++){
    arr.push(nums[i])
     for(let j=i+1;j<nums.length;j++){
       arr.push(arr[arr.length-1]+nums[j])
    }
   }
  //  console.log(arr)
   sortMerge(arr,0,arr.length-1);
   let ans=0;
   for(let i=left-1;i<=right-1;i++){
     ans+=arr[i]
   }
   return ans % (1e9+7);

};
function sortMerge(num,l,r){
  if(l>=r)return;
  let mid=Math.floor((l+r)/2);
  let p1=l;let p2=mid+1;
  sortMerge(num,l,mid)
  sortMerge(num,mid+1,r)
  let k=0;
  let temp=new Array(r-l+1).fill(0)
  while(p1<=mid||p2<=r){
    if(p2>r||(p1<=mid&&num[p1]<=num[p2])){
      temp[k++]=num[p1++]
    }else{
      temp[k++]=num[p2++]
    }
  }
  for(let i=0;i<temp.length;i++)num[l+i]=temp[i]
  return ;
}