欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-03-05 13:30:00
...

题目描述

在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
输入描述:
题目保证输入的数组中没有的相同的数字

数据范围:

对于%50的数据,size<=10^4

对于%75的数据,size<=10^5

对于%100的数据,size<=2*10^5

思路

利用归并排序,在合并时同时计算逆序个数。

public class Solution {

    private long count = 0;
    private int[] tmp;

    public int InversePairs(int [] array) {
        int n = array.length;
        tmp = new int[n];
        mergeSort(array, 0, n - 1);
        return (int)count % 1000000007;
    }

    public void mergeSort(int[] arr, int left, int right) {
        if (right - left < 1) return;
            int mid = left + (right - left) / 2;
            mergeSort(arr, left, mid);
            mergeSort(arr, mid + 1, right);
            merge(arr, left, mid, right);

    }

    public void merge(int[] arr, int left, int mid, int right) {

        int i = left;
        int j = mid + 1;
        int t = 0;

        while (i <= mid && j <= right) {
            if (arr[i] > arr[j]) {
                count += (mid - i + 1);
                if (count > 1000000007) 
                    count %= 1000000007;
                tmp[t++] = arr[j++];
            } else {
                tmp[t++] = arr[i++];
            }
        }

        while (i <= mid) tmp[t++] = arr[i++];
        while (j <= right) tmp[t++] = arr[j++];


        int k = 0;
        while (left <= right) {
            arr[left++] = tmp[k++];
        }
    }
}