【劍指offer】數組中的逆序對，C++實現

原創博文，轉載請注明出處！
本題牛客網地址

博客文章索引地址

博客文章中代碼的github地址

1.題目

2.思路

3.代碼

class Solution {
public:
    int InversePairs(vector<int> data) {
        if(data.size() == 0){
            return 0;
        }
        // 排序的輔助數組
        vector<int> copy;
        for(int i = 0; i < data.size(); ++i){
            copy.push_back(data[i]);
        }
        return InversePairsCore(data, copy, 0, data.size() - 1) % 1000000007;
    }
    long InversePairsCore(vector<int> &data, vector<int> &copy, int begin, int end){
        // 如果指向相同位置，則沒有逆序對。
        if(begin == end){
            copy[begin] = data[end];
            return 0;
        }
        // 求中點
        int mid = (end + begin) >> 1;
        // 使data左半段有序，並返回左半段逆序對的數目
        long leftCount = InversePairsCore(copy, data, begin, mid);
        // 使data右半段有序，並返回右半段逆序對的數目
        long rightCount = InversePairsCore(copy, data, mid + 1, end);

        int i = mid; // i初始化為前半段最后一個數字的下標
        int j = end; // j初始化為后半段最后一個數字的下標
        int indexcopy = end; // 輔助數組復制的數組的最后一個數字的下標
        long count = 0; // 計數，逆序對的個數，注意類型

        while(i >= begin && j >= mid + 1){
            if(data[i] > data[j]){
                copy[indexcopy--] = data[i--];
                count += j - mid;
            }
            else{
                copy[indexcopy--] = data[j--];
            }
        }
        for(;i >= begin; --i){
            copy[indexcopy--] = data[i];
        }
        for(;j >= mid + 1; --j){
            copy[indexcopy--] = data[j];
        }
        return leftCount + rightCount + count;
    }
};