剑指offer之数组中的逆序对

题目描述

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007

输入描述

题目保证输入的数组中没有的相同的数字
数据范围：
对于%50的数据,size<=10^4
对于%75的数据,size<=10^5
对于%100的数据,size<=2*10^5

输入

1,2,3,4,5,6,7,0

输出

7

解题思路

菜鸡本菜依然是只能想出暴力的路线，尝试其他想法，无果。
参考书上的思路，采用了归并排序的思想，结果注意取模：

Code

• 书上的代码

class Solution {
public:
    int InversePairs(vector<int> data) {
        if(!data.size()) {
            return 0;
        }
        vector<int> copy;
        for(int i = 0; i<data.size(); i++) {
            copy.push_back(data[i]);
        }
        int count = InversePairsCore(data, copy, 0, data.size()-1);
        return count;
    }
    
    int InversePairsCore(vector<int> &data, vector<int> &copy, int start, int end) {
        if(start == end) {
            copy[start] = data[start];
            return 0;
        }
        
        int length = (end-start)/2;
        int left = InversePairsCore(copy, data, start, start+length);
        int right = InversePairsCore(copy, data, start+length+1, end);
        
        //i初始化为前半段最后一个数字的下标
        int i = start+length;
        // j初始化为后半段最后一个数字的下标
        int j = end;
        int indexCopy = end;
        int count = 0;
        while(i>=start && j>=start+length+1) {
            if(data[i] > data[j]) {
                copy[indexCopy--] = data[i--];
                count += j-start-length;
            } else {
                copy[indexCopy--] = data[j--];
            }
            count = count % 1000000007;
        }
        
        for(; i>=start; i--) {
            copy[indexCopy--] = data[i];
        }
        for(; j>=start+length+1; j--) {
            copy[indexCopy--] = data[j];
        }
        
        return (left+right+count)%1000000007;
    }
};

• java

public class Solution {
    private int result = 0;
    public void merge(int[] array, int left, int right) {
        int mid = (left+right)/2;
        int p1 = left, p2 = mid+1;
        int[] temp = new int[right-left+1];
        int index = 0, mod = 1000000007;
        while(p1 <= mid && p2 <= right) {
            if(array[p1] <= array[p2]) {
                temp[index++] = array[p1++];
            } else {
                temp[index++] = array[p2++];
                result = (result+mid-p1+1)%mod;
            }
        }
        while(p1 <= mid) {
            temp[index++] = array[p1++];
        }
        while(p2 <= right) {
            temp[index++] = array[p2++];
        }
        for(int i = 0; i < index; i++) {
            array[left+i] = temp[i];
        }
    }
    public void mergeSort(int[] array, int left, int right) {
        if(left < right) {
            int mid = (left+right)/2;
            mergeSort(array, left, mid);
            mergeSort(array, mid+1, right);
            merge(array, left, right);
        }
    }
    public int InversePairs(int [] array) {
        mergeSort(array, 0, array.length-1);
        return result;
    }
}

总结

二刷终于想起了归并，然而巨菜想着非递归归并去count，结果处理不要尾巴。。（即只能处理2的次方的长度）然后看了评论区的递归版本，真香