【剑指offer】51数组中的逆序对

题目描述

在一个数组中，左边的数如果比右边的数大，则折成两个数构成一个逆序对，请打印所有逆序对。

输入  1,2,3,4,5,6,7,0
输出  7
[1,0] [2,0] [3,0] [4,0]...[7,0]

思路

这和【小和问题】一样属于【归并排序】类的问题，理解了归并排序就可以解。

 static int findPairs(int[] arr, int start, int end) {
        if (start == end || arr.length < 2) return 0;

        int mid = start + (end - start) / 2;

        return findPairs(arr, start, mid)+ findPairs(arr, mid + 1, end) + merge(arr, start, mid, end);
    }

static int merge(int[] arr, int start, int mid, int end) {
    int[] temp = new int[end - start + 1];
    int p1 = start, p2 = mid + 1;
    int index = 0, count = 0;

    while (p1 <= mid && p2 <= end) {
        count += arr[p1] > arr[p2] ? end - p2 + 1 : 0;
        temp[index++] = arr[p1] > arr[p2] ? arr[p1++] : arr[p2++];
    }
    while (p1 <= mid) temp[index++] = arr[p1++];
    while (p2 <= end) temp[index++] = arr[p2++];

    for (int i = 0; i < temp.length; i++) {
        arr[start + i] = temp[i];
    }
    return count;
}

public static void main(String[] args) {
    int[] arr = new int[]{1, 2, 3, 4, 5, 6, 7, 0};
    int pairs = findPairs(arr, 0, arr.length - 1);
    System.out.println(pairs);
    for(int num : arr){
        System.out.print(num + " ");
    }
}