剑指offer--逆序对

描述

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

 

思路：归并排序，元素移动的单位和就是逆序对的个数

class Solution {
public:
    int count=0,MAX=1000000007;
    // 采用归并排序，其交换次数就是逆序对的个数
    int InversePairs(vector<int> data) {
        sort(data,0,data.size()-1);
        return count;
    }
    
    vector<int> sort(vector<int>& init,int start,int end){
        if(start==end){return vector<int>(1,init[start]);}
        int mid=(start+end)/2;
        vector<int> a=sort(init,start,mid);
        vector<int> b=sort(init,mid+1,end);
        return merge(a,b);
    }
    
    vector<int> merge(vector<int> a,vector<int> b){
        vector<int> tmp;
        int i=0,j=0;
        while(i<a.size() && j<b.size()){
            if(a[i]<=b[j]){tmp.push_back(a[i++]);}
            else{tmp.push_back(b[j++]);count=(count+a.size()-i)%MAX; }
        }
        if(i==a.size()){   // 这个if和else if其实不需要
            while(j<b.size()){
                tmp.push_back(b[j++]);
            }
        }else if(j==b.size()){
            while(i<a.size()){
                tmp.push_back(a[i++]);
            }
        }
        return tmp;
    }
};