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

这道题归根结底就是一个归并问题,逆序对本质上就是比较大小,如果两边作为一个整体比较过那么就可以排序合并(因为这个过程每一步都计算了count的值,所以合并起来是可以的)。

下面的k应该是mid+1(从中间的右边一位开始取,不能等于)。

剑指 Offer 51: 数组中的逆序对_第1张图片

剑指 Offer 51: 数组中的逆序对_第2张图片

临时的数组十分重要,它的范围在每个循环的left到数组结尾的tmpK

 

剑指 Offer 51: 数组中的逆序对_第3张图片

class Solution {
    int count;
    public int reversePairs(int[] nums) {
        this.count = 0;
        int[] tmp = new int[nums.length];
        mergeSort(nums,0,nums.length-1,tmp);
        return count;
    }
    public void merge(int[] nums, int left, int mid, int right, int[] tmp){
        int tmpK = 0, low = left, k = mid + 1;
        while(low <= mid && k <= right){//当两边都还有值
            if(nums[low] <= nums[k])
                tmp[tmpK++] = nums[low++];
            else{
                tmp[tmpK++] = nums[k++];
                // count++;一个数大于其说明后面的都大于
                //只考虑从左往右的,大于直接左边右移,不用一个个加
                count = count + mid - low + 1;
            }
        }
        while(low <= mid)
            tmp[tmpK++] = nums[low++];
        while(k <= right)//前面可以取mid,这里却不能取right
            tmp[tmpK++] = nums[k++];
        for(int i = 0; i < tmpK; i++){//起始位置right到tmpK
            nums[left + i] = tmp[i];
        }
    }
    public void mergeSort(int[] nums,int left,int right,int [] tmp){
        if(left >= right)
            return ;
        int mid = (left + right)/2;
        mergeSort(nums,left,mid,tmp);
        mergeSort(nums,mid+1,right,tmp);
        merge(nums,left,mid,right,tmp);
    }
}

你可能感兴趣的:(#,剑指offer,算法,leetcode,数据结构)