剑指offer：数组中的逆序对（分治）

 

[编程题]数组中的逆序对

• 热度指数：243682 时间限制：2秒 空间限制：32768K
• 算法知识视频讲解

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

输入描述:

题目保证输入的数组中没有的相同的数字

数据范围：

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

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

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

 

示例1

输入

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

输出

7

 

思路：首先可以O(nlogn)的离散化+树状数组解决，但是这题卡常数就算了。改用分治来做也是O(nlogn)而且常数很小，具体就是类似归并排序的过程。

class Solution {
public:
    const int mod = 1e9+7;
    int ans = 0, *tmp;
    vector<int>da;
    void add(int x){
        ans += x;
        if(ans >= mod) ans -= mod;
    }
    void cdq(int l, int r){
        if(l == r) return;
        int mid = l+r>>1,cnt=0, j=mid;
        cdq(l, mid);
        cdq(mid+1, r);
        for(int i=l; i<=mid; ++i){
            while(j+1<=r && da[i]>da[j+1]) ++j,tmp[cnt++]=da[j];
            add(j-mid);
            tmp[cnt++] = da[i];
        }
        while(j<r) tmp[cnt++] = da[++j];
        for(int i=l; i<=r; ++i){
            da[i] = tmp[i-l];
        }
    }
    int InversePairs(vector<int> data) {
        da = data;
        int len = data.size();
        tmp = new int[len+3];
        cdq(0, len-1);
        return ans;
    }
};