一开始就联想到了用归并排序求逆序数这道题,可是当时求逆序数的时候,不需要知道逆序数出现在哪个位置,只用记录逆序数个数即可,所以改了一下当时的代码,利用index数组记录某位置的元素的初始位置,再通过result[index[pos]] += 1
即可把当前逆序数加入到原位置的结果。然而我写的加入到原位置的代码会超时。
超时代码:
class Solution:
def countSmaller(self, nums: List[int]) -> List[int]:
l = len(nums)
if l == 0:
return []
result = [0 for i in range(l)]
# 需要用此得到某位置元素最后挪到的位置
# 对应关系:结果位置->初始位置
index = [i for i in range(l)]
# 归并排序
def Sort(left, right):
if left < right:
mid = (left+right) >> 1
# print(left, mid, right)
Sort(left, mid)
Sort(mid+1, right)
Merge(left, mid, right)
def Merge(left, mid, right):
tmp = []
tmp_index = [0 for i in range(right-left+1)]
i = left
j = mid+1
k = 0
while i <= mid and j <= right:
if nums[i] > nums[j]:
tmp.append(nums[j])
# 逆序数更新,只要出现一个逆序数,那么就更新
for pos in range(i, mid+1):
result[index[pos]] += 1
tmp_index[k] = index[j]
j += 1
else:
tmp.append(nums[i])
tmp_index[k] = index[i]
i += 1
k += 1
while i <= mid:
tmp.append(nums[i])
tmp_index[k] = index[i]
k += 1
i += 1
while j <= right:
tmp.append(nums[j])
tmp_index[k] = index[j]
k += 1
j += 1
nums[left:right+1] = tmp[:]
index[left:right+1] = tmp_index[:]
Sort(0, l-1)
return result
后来改了一下逆序数的更新方式。将某个位置的逆序数积攒在一起更新,而不是有一个该位置逆序数就更新,改进方式是将更新逆序数放在下面代码中的位置,具体看注释。需要注意的是增加的逆序数个数需要自己模拟一下程序过程,推一下。
可通过代码:
class Solution:
def countSmaller(self, nums: List[int]) -> List[int]:
l = len(nums)
if l == 0:
return []
result = [0 for i in range(l)]
# 需要用此得到某位置元素最后挪到的位置
# 对应关系:结果位置->初始位置
index = [i for i in range(l)]
# 归并排序
def Sort(left, right):
if left < right:
mid = (left+right) >> 1
# print(left, mid, right)
Sort(left, mid)
Sort(mid+1, right)
Merge(left, mid, right)
def Merge(left, mid, right):
tmp = []
tmp_index = [0 for i in range(right-left+1)]
i = left
j = mid+1
k = 0
while i <= mid and j <= right:
if nums[i] > nums[j]:
tmp.append(nums[j])
tmp_index[k] = index[j]
j += 1
else:
# 找到该位置此次排序的所有逆序数,更新
result[index[i]] += j-mid-1
tmp.append(nums[i])
tmp_index[k] = index[i]
i += 1
k += 1
while i <= mid:
tmp.append(nums[i])
tmp_index[k] = index[i]
# 找到该位置此次排序的所有逆序数,更新
result[index[i]] += (right-mid)
k += 1
i += 1
while j <= right:
tmp.append(nums[j])
tmp_index[k] = index[j]
k += 1
j += 1
nums[left:right+1] = tmp[:]
index[left:right+1] = tmp_index[:]
Sort(0, l-1)
return result