给你一个整数数组 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]
解法一:二分查找+重新构建一个新数组
时间复杂度nlogn,但是力扣上超时了,最新的解法是树状数组才能ac
思路:
思路是将给定数组从最后一个开始,用二分法插入到一个新的数组new_arr,
这样新数组new_arr就是有序的,
那么此时该数字在新数组中的坐标就是原数组中其右边所有较小数字的个数
class Solution(object):
def countSmaller(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
# 使用用二分搜索法,
# 思路是将给定数组从最后一个开始,用二分法插入到一个新的数组,
# 这样新数组就是有序的,
# 那么此时该数字在新数组中的坐标就是原数组中其右边所有较小数字的个数
n = len(nums)
ans = [0] * n
new_arr = []
for i in range(n-1,-1,-1):
left = 0
right = len(new_arr)
while left < right:
mid = left + (right - left) // 2
if nums[i] > new_arr[mid]:
left = mid + 1
else:
right = mid
ans[i] = left
# print(left)
if left >= len(new_arr):
new_arr.append(nums[i])
else:
new_arr = new_arr[0:left] + [nums[i]] + new_arr[left:]
# print(new_arr)
return ans
解法二:树状数组
挖坑中。。。。。。