题目
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
示例 1:
输入: [7,5,6,4] 输出: 5
限制:
0 <= 数组长度 <= 50000
分析
该题第一种思路是采用冒泡排序,排序过程中每交换一对元素就说明这对元素构成逆序对,所以统计冒泡排序一共交换多少次就相当于统计了有多少个逆序对。
但是用冒泡排序它的时间复杂度为O(n ^2),空间复杂度为O(1)。
第二种思路就是采用归并排序,时间复杂度为O(nlgn),空间复杂度为O(n)。
例如下图,我们要求这个数组的逆序对个数,将数组划分为一半,左半部分右半部分假设先都用冒泡排序来计算这两边的逆序对个数x1和x2,x3也就是整个数组的逆序对该怎么求呢?
我们可以先定义一个辅助数组temp,并看下图定义好left,right,mid指针,以及i,j指针,i,j都是从left和mid+1开始的,然后比较左右两部分的数组元素大小,如果nums[i]<=nums[j],将数组nums[i]存入temp数组中,如果nums[i]>nums[j],将数组nums[j]存入temp数组中并且说明下标i到下标mid的所有元素都大于nums[j],也就说明它们都构成逆序对,统计这部分逆序对总数,其实也就是(mid+1)-i,比如再下张图中数组中新加的2,j指向这个2时统计的逆序对同样是(mid+1)-i。最后别忘了把临时数组temp赋值给原数组。
之前解释的归并排序中的那个冒泡排序其实是不需要的,只需要用递归就可以解决,刚才说的冒泡排序其实是为了方便解释,因为x1,x2同样可以是归并排序的子问题。
代码
//冒泡排序法
class Solution {
public int reversePairs(int[] nums) {
int k = 0;
for(int i = 0;i < nums.length - 1;i++){
for(int j = 0;j < nums.length - i - 1;j++){
if(nums[j] > nums[j+1]){
k++;
int temp = nums[j];
nums[j] = nums[j+1];
nums[j+1] = temp;
}
}
}
return k;
}
}
//归并排序
class Solution {
public int reversePairs(int[] nums) {
if(nums == null||nums.length==0){
return 0;
}
return mergeSort(nums,0,nums.length-1);
}
int mergeSort(int[] nums,int left,int right){
if(left>=right){
return 0;
}
int mid = (left+right)/2;
int x1 = mergeSort(nums,left,mid);
int x2 = mergeSort(nums,mid+1,right);
int x3 = merge(nums,left,mid,mid+1,right);
return x1+x2+x3;
}
int merge(int[] nums, int l1, int r1, int l2, int r2){
int[] temp = new int[r2-l1+1];
int count = 0;
int i = l1, j = l2, k = 0;
while(i<=r1&&j<=r2){
if(nums[i]>nums[j]){
count = count+(l2-i);
temp[k++] = nums[j++];
}else{
temp[k++] = nums[i++];
}
}
while(i<=r1) temp[k++] = nums[i++];
while(j<=r2) temp[k++] = nums[j++];
// 把临时数组复制回原数组
k = 0;
for(i = l1; i <= r2; i++){
nums[i] = temp[k++];
}
return count;
}
}