剑指 Offer 51. 数组中的逆序对
难度困难427
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
示例 1:
输入: [7,5,6,4]
输出: 5
限制:
0 <= 数组长度 <= 50000
解析:
归并排序, 如果mid之后的数排到前面了,那么re+=这个数原来位置之前的比他大的数的个数。模板题。
时间复杂度O(nlogn) 空间复杂度 O(n)
代码:
class Solution {
static int N = 50005;
int[] tmp = new int[N];
int res = 0;
public int reversePairs(int[] nums) {
// int res = merge_sort(nums, 0, nums.length - 1);
return merge_sort(nums, 0, nums.length - 1);
}
int merge_sort(int[] nums, int l, int r){
if(l >= r) return 0;
int mid = l + r >> 1;
merge_sort(nums, l, mid);
merge_sort(nums, mid + 1, r);
int k = 0, i = l, j = mid + 1;
while(i <= mid && j <= r){
if(nums[i] <= nums[j]) tmp[k++] = nums[i++];
else{
tmp[k++] = nums[j++];// mid后的数排进了mid前 所以就可以统计逆序对, 说明 mid-i+1 个数比这个数大,却排在了这个数之前
res += mid - i + 1;
}
}
//右边的排完了
while(i <= mid) tmp[k++] = nums[i++];
// 左边的排完了
while(j <= r) tmp[k++] = nums[j++];
//物归原主,赋值给原数组
for(i = l, j = 0; i <= r; i++, j++){
nums[i] = tmp[j];
}
return res;
}
}