LeetCode 315. 计算右侧小于当前元素的个数(Count of Smaller Numbers After Self)

题目描述:

给定一个整数数组 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;

};


猜你喜欢

转载自blog.csdn.net/cc_fys/article/details/80770248