题目描述:
给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i]
的值是 nums[i]
右侧小于 nums[i]
的元素的数量。
示例:
输入: [5,2,6,1]
输出: [2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1).
2 的右侧仅有 1 个更小的元素 (1).
6 的右侧有 1 个更小的元素 (1).
1 的右侧有 0 个更小的元素.
思路:
这道题用暴力解法显然是很简单的,但是用暴力解时间复杂度是O(n
2)显然是不满足要求的。换一种思路,从后往前遍历数组,将每次遍历过后的数放入另外一个数组temp中,使其保持有序。在计算右侧小于当前元素的个数,用二分查找的方式查找temp这个有序array,那么在数组左边的数即是小于当前元素的数,其位置index既为个数信息。
JavaScript:
/**
* @param {number[]} nums
* @return {number[]}
*/
var countSmaller = function(nums) {
//temp升序
var temp=[];
var result=[];
for(var i=nums.length-1;i>=0;i--)
{
//console.log(temp);
var left=0;
var right=temp.length;
while(left<right)
{
var mid=left+parseInt((right-left)/2);
if(temp[mid]>=nums[i])
right=mid;
else
left=mid+1;
}
result[i]=left;
temp.splice(left,0,nums[i]);
}
return result;
};