LeetCode面试51:数组中的逆序对
题目描述:
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
样例输入及输出:
输入: [7,5,6,4]
输出: 5
代码示例与讲解:
//====================
//解法1,在数组很长时会超时
class Solution01{
static public int reversePairs(int[] nums) {
int revnums = 0;
for (int i = 0; i < nums.length; i++) {
for (int j = i+1; j <nums.length; j++) {
if (nums[j]<nums[i]) {
revnums++;
}
}
}
return revnums;
}
}
//==================================================
//解法2:
// 核心思想是两个有序序列进行合并时可以快速计算得到逆序对数,因此考虑排序方法
// 在已有的排序中归并排序可以很好地反应元素之间的大小关系
class Solution{
static public int reversePairs(int[] nums){
int len = nums.length;
if (len<2) {
return 0;
}
int[] copy = new int[len];
for (int i = 0; i < len; i++) {
copy[i] = nums[i];
}
int[]temp = new int[len];
return reversePairs(copy,0,len-1,temp);
}
private static int reversePairs(int[] nums, int left, int right, int[] temp) {
if(left==right){
return 0;
}
//注意:数组中逆序对的总数量 = 左半部分中逆序对数 + 右半部分中逆序对数 + 左右均有序后的交叉逆序对数
//注意根据表达式,当数组中只有两个元素时,mid = left
int mid = left+(right-left)/2;
int leftpairs = reversePairs(nums,left,mid,temp);
int rightpairs = reversePairs(nums,mid+1,right,temp);
//若左右部分本来就处于有序状态
if (nums[mid]<=nums[mid+1]){
return leftpairs+rightpairs;
}
//通过归并排序计算交叉逆序数
int crosspairs = mergeAndCount(nums,left,mid,right,temp);
return leftpairs+rightpairs+crosspairs;
}
//进行归并排序并且记录逆序对数量
private static int mergeAndCount(int[] nums, int left, int mid, int right, int[] temp) {
//复制数组
for (int i = left; i <= right; i++) {
temp[i] = nums[i];
}
//i与j分别是左右两部分数组的起点
int i = left;
int j = mid + 1;
int count = 0;
for (int k = left; k <= right; k++) {
//前两种情况对应左、右半部分已经排序结束
if (i == mid + 1) {
nums[k] = temp[j];
j++;
} else if (j == right + 1) {
nums[k] = temp[i];
i++;
} else if (temp[i] <= temp[j]) {
nums[k] = temp[i];
i++;
} else {
nums[k] = temp[j];
j++;
count += (mid - i + 1);
}
}
return count;
}
}
public class LeetCode_51 {
public static void main(String[] args) {
int[] test = new int[]{7,5,6,4};
int n = Solution.reversePairs(test);
System.out.println(n);
}
}