数组中的逆序对
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
示例 1:
输入: [7,5,6,4]
输出: 5
暴力法
时间复杂度:O(N^2)
将所有可能枚举出来,如果满足逆,则 cnt+1,最后返回 cnt 即可
归并排序
我们进行一次归并排序,并在归并过程中计算逆序数,换句话说 逆序对是归并排序的副产物。
归并排序实际上会把数组分成两个有序部分,我们不妨称其为左和右,归并排序的过程中会将左右两部分合并成一个有序的部分,对于每一个左右部分,我们分别计算其逆序数,然后全部加起来就是我们要求的逆序数。
比如对于左:[1,2,3,4]右:[2,5]。 其中 i,j 指针如粗体部分。 那么 逆序数就是 mid - i + 1 也就是 3 - 2 + 1 = 2 即(3,2)和 (4,2)。 其原因在于如果 3 大于 2,那么 3 后面不用看了,肯定都大于 2。
class Solution:
def reversePairs(self, nums: List[int]) -> int:
self.cnt = 0
self.mergesort(nums, 0, len(nums) - 1, [])
return self.cnt
# 归并两个有序的数组,归并过程统计逆序对
def merge(self, nums, start, mid, end, temp):
i = start
j = mid + 1
while i <= mid and j <= end:
if nums[i] <= nums[j]:
temp.append(nums[i])
i += 1
else:
self.cnt += mid + 1 - i
temp.append(nums[j])
j += 1
while i <= mid:
temp.append(nums[i])
i += 1
while j <= end:
temp.append(nums[j])
j += 1
# 把归并好的结果放回原数组对应位置
for i in range(len(temp)):
nums[start + i] = temp[i]
temp.clear()
# 归并排序
def mergesort(self, nums, start, end, temp):
if start >= end:
return
mid = (start + end) // 2
self.mergesort(nums, start, mid, temp)
self.mergesort(nums, mid + 1, end, temp)
self.merge(nums, start, mid, end, temp)