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

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

题目描述

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

解答

class Solution {

    // 将数组划分两段进行归并排序（按从小到大排），确保前后两段都是有序的
    // 在排序过程中当后面那一段的某个元素被选中时，则前面那段中剩下的元素都可以和当前元素组成逆序对，
    // 依此类推，在排序的过程中可计算得到数组中逆序对的总数
    // 我们可以将数组拆分成每段只含一个元素的有序段，这样然后回溯进行排序，保证了算法的可执行性

    public int reversePairs(int[] nums) {
        if(nums.length < 2){
            return 0;
        }

        int[] copy = new int[nums.length];  //待排序的数组
        for(int i = 0;i < nums.length;i++){
            copy[i] = nums[i];
        }

        int[] temp = new int[nums.length];  //辅助数组
        return reversePair(copy,0,copy.length-1,temp);

    }


    public int reversePair(int[] nums,int left,int right,int[] temp){
        if(left == right){
            return 0;
        }
        int mid = (left + right)/2;
        int leftNum = reversePair(nums,left,mid,temp);
        int rightNum = reversePair(nums,mid+1,right,temp);

        int currNum = mergeSort(nums,left,mid,right,temp);

        System.out.println("leftNum:"+leftNum +" rightNum:"+ rightNum +" currNum:"+ currNum);

        return leftNum + rightNum + currNum;
    }
    
    public int mergeSort(int[] nums,int left,int mid,int right,int[] temp){
        int sum = 0;

        for(int i = left;i <= right ;i++){
            temp[i] = nums[i];
        }

        int i = left;
        int j = mid+1;
        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++;
                sum += (mid-i+1);
            }
        }

        return sum;
    }
}