leetcode 315. 计算右侧小于当前元素的个数
问题描述: 给你一个整数数组 nums **,按要求返回一个新数组 counts **。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
示例 1:
输入: nums = [5,2,6,1]
输出: [2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
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
}