剑指Offer——数组中的逆序对

剑指Offer——数组中的逆序对

题意

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组，求出这个数组中的逆序对的总数。

分析

先离散化，再利用树状数组。首先先计算出每个数最后排完序所在的位置，用一个map来存储。然后动态加点，每次把每个数该在的位置上加上1，若一个数该在的位置为x，则在x位置加上1。用query(x)查询1-x的和，也就是之前（包括自己）有多少比自己小的数，则i-query(x)就是比自己大的数。

代码

class Solution {
public:
    int n,ans;
    int BIT[50005];
    inline int lowbit(int x) {
        return x&-x;
    }
    void update(int p, int x) {
        while(p<=n) {
            BIT[p]+=x;
            p+=lowbit(p);
        }
    }
    int query(int p) {
        int ret=0;
        while(p) {
            ret+=BIT[p];
            p-=lowbit(p);
        }
        return ret;
    }
    int reversePairs(vector<int>& nums) {
        memset(BIT,0,sizeof(BIT));
        n=nums.size(); ans=0;
        vector<int> a(nums.begin(), nums.end());
        sort(a.begin(), a.end());
        map<int, int> M;
        for (int i=0; i<n; ++i)  M[a[i]]=i+1;
        for (int i=0; i<n; ++i) {
            update(M[nums[i]], 1);
            ans+=((i+1)-query(M[nums[i]]));
        }
        return ans;
    }
};