归并排序的思想解决逆序对和翻转对问题

剑指 Offer 51. 数组中的逆序对

在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。

示例 1:
输入: [7,5,6,4]
输出: 5

代码如下:

//通用解法: 归并排序。时间复杂度高。
class Solution {
    public int reversePairs(int[] nums) {
        if (nums == null || nums.length == 0) return 0;
        return mergSort(nums,0,nums.length-1);
    }
    public int mergSort(int[] array,int left,int right) {
        if (left == right) {
            return 0;
        }
        int mid = (left +right) /2;
        return mergSort(array,left,mid)+mergSort(array,mid+1,right)+merge(array,left,mid,right);
    }
    public int merge(int[] array,int left,int mid,int right) {
       int[] tmpArr = new int[right-left+1];
        int num = 0; // 翻转对数量
        int i = left;
        int j = mid+1;
        while (i <= mid) {
            // 此时(i,j)是符合条件的,j在右组中继续向右滑动,继续向右扩,看在此时i不动的情况下,j最多能向右扩到哪里
            while (j <= right && array[i] > array[j]) { // 【注意】:num为整型范围,乘以2可能溢出
                j++;
            }
            // 此时,j来到右组中第一个不满足条件的位置,找到了在此时i的情况下,能产生的翻转对数量:
            num += (j - mid - 1);
            // 此时的i位置已经结算完答案,i向右滑动一个位置,尝试下一个i,能产生多少个翻转对
            i++;
        }
        int s1 = left;
        int s2 = mid+1;
        int k =0;
        while (s1 <= mid && s2 <= right) {
            if (array[s1] <= array[s2]) {
                tmpArr[k++] = array[s1++];
            }else {
                tmpArr[k++] = array[s2++];
            }
        }
        while (s1 <= mid) {
            tmpArr[k++] = array[s1++];
        }
        while (s2 <= right) {
            tmpArr[k++] = array[s2++];
        }
        for (int h = 0; h < tmpArr.length; h++) {
            array[left++] = tmpArr[h];
        }
        return num;
    }
}

解法二:归并排序思想:
class Solution {
    public int reversePairs(int[] nums) {
        if (nums == null || nums.length == 0) return 0;
        return mergSort(nums,0,nums.length-1);
    }
    public int mergSort(int[] array,int left,int right) {
        if (left == right) {
            return 0;
        }
        int mid = (left +right) /2;
        return mergSort(array,left,mid)+mergSort(array,mid+1,right)+merge(array,left,mid,right);
    }
    public int merge(int[] array,int left,int mid,int right) {
       int[] tmpArr = new int[right-left+1];
        int num = 0; // 翻转对数量
        int s1 = left;
        int s2 = mid+1;
        int k =0;
        while (s1 <= mid && s2 <= right) {
            if (array[s1] <= array[s2]) {
                tmpArr[k++] = array[s1++];
            }else {
                num += mid -s1 +1;
                tmpArr[k++] = array[s2++];
            }
        }
        while (s1 <= mid) {
            tmpArr[k++] = array[s1++];
        }
        while (s2 <= right) {
            tmpArr[k++] = array[s2++];
        }
        for (int h = 0; h < tmpArr.length; h++) {
            array[left++] = tmpArr[h];
        }
        return num;
    }
}
493. 翻转对

给定一个数组 nums ,如果 i < j 且 nums[i] > 2*nums[j] 我们就将 (i, j) 称作一个重要翻转对。

你需要返回给定数组中的重要翻转对的数量。

示例 1:
输入: [1,3,2,3,1]
输出: 2
示例 2:
输入: [2,4,3,5,1]
输出: 3
注意:
给定数组的长度不会超过50000。
输入数组中的所有数字都在32位整数的表示范围内。

代码如下:

class Solution {
    public int reversePairs(int[] nums) {
        if (nums == null || nums.length == 0) return 0;
        return mergSort(nums,0,nums.length-1);
    }
    public int mergSort(int[] array,int left,int right) {
        if (left == right) {
            return 0;
        }
        int mid = (left +right) /2;
        return mergSort(array,left,mid)+mergSort(array,mid+1,right)+merge(array,left,mid,right);
    }
    public int merge(int[] array,int left,int mid,int right) {
       int[] tmpArr = new int[right-left+1];
        int num = 0; // 翻转对数量
        int i = left;
        int j = mid+1;
        while (i <= mid) {
            // 此时(i,j)是符合条件的,j在右组中继续向右滑动,继续向右扩,看在此时i不动的情况下,j最多能向右扩到哪里
            while (j <= right && array[i] > (long)2*array[j]) { // 【注意】:num为整型范围,乘以2可能溢出
                j++;
            }
            // 此时,j来到右组中第一个不满足条件的位置,找到了在此时i的情况下,能产生的翻转对数量:
            num += (j - mid - 1);
            // 此时的i位置已经结算完答案,i向右滑动一个位置,尝试下一个i,能产生多少个翻转对
            i++;
        }
        int s1 = left;
        int s2 = mid+1;
        int k =0;
        while (s1 <= mid && s2 <= right) {
            if (array[s1] <= array[s2]) {
                tmpArr[k++] = array[s1++];
            }else {
                tmpArr[k++] = array[s2++];
            }
        }
        while (s1 <= mid) {
            tmpArr[k++] = array[s1++];
        }
        while (s2 <= right) {
            tmpArr[k++] = array[s2++];
        }
        for (int h = 0; h < tmpArr.length; h++) {
            array[left++] = tmpArr[h];
        }
        return num;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
归并排序是一种分治算法,它的基本思想是将一个大问题分解成若干个小问题,然后递归地解决这些小问题,最后将它们合并成一个整体的解。在归并排序中,我们可以通过在合并两个有序数组的过程中统计逆序对的数量来求解逆序对。 具体来说,我们可以将原数组分成左右两个子数组,然后递归地对左右两个子数组进行排序,并统计左右两个子数组内部的逆序对数量。接着,我们再将左右两个子数组合并成一个有序数组,并统计左右两个子数组之间的逆序对数量。最后,我们将左右两个子数组的逆序对数量加起来,就得到了原数组的逆序对数量。 下面是使用归并排序逆序对的代码实现: ```python def merge_sort(arr): if len(arr) <= 1: return arr, 0 mid = len(arr) // 2 left, left_count = merge_sort(arr[:mid]) right, right_count = merge_sort(arr[mid:]) merged, merge_count = merge(left, right) return merged, left_count + right_count + merge_count def merge(left, right): i, j = 0, 0 count = 0 merged = [] while i < len(left) and j < len(right): if left[i] <= right[j]: merged.append(left[i]) i += 1 else: merged.append(right[j]) j += 1 count += len(left) - i merged += left[i:] merged += right[j:] return merged, count ``` 其中,`merge_sort` 函数用于对数组进行归并排序,并返回排序后的数组和逆序对数量。`merge` 函数用于合并两个有序数组,并统计它们之间的逆序对数量。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值