[路飞]_js算法:leetcode 315-计算右侧小于当前元素的个数

111 阅读1分钟

leetcode 315. 计算右侧小于当前元素的个数

问题描述: 给你一个整数数组 nums **,按要求返回一个新数组 counts **。数组 counts 有该性质: counts[i] 的值是  nums[i] 右侧小于 nums[i] 的元素的数量。

示例 1:

输入: nums = [5,2,6,1]
输出: [2,1,1,0] 
解释: 
5 的右侧有 2 个更小的元素 (21)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素

示例 2:

输入: nums = [-1]
输出: [0]

示例 3:

输入: nums = [-1,-1]
输出: [0,0]

思路: map+归并排序(从大到小)

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var countSmaller = function(nums) {
 let arr=[];
  for(let i=0;i<nums.length;i++)arr.push({num:nums[i],index:i,cnt:0})
  sortMerge(arr,0,arr.length-1)
  let res=new Array(arr.length).fill(0);
  for(let i=0;i<arr.length;i++)res[arr[i].index]=arr[i].cnt
  return res;

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