面试题51. 数组中的逆序对
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。
示例 1:
输入: [7,5,6,4]
输出: 5
限制:
0 <= 数组长度 <= 50000
思路
离散化 + 反向构建树状数组
实现代码
class BIT{
private:
vector<int> tree;
int len;
public:
BIT(int n):len(n), tree(n+1,0){}
static int lowbit(int x){
return x & -x;
}
int query(int x){
int res = 0;
for(int i = x; i > 0; i -= lowbit(i)){
res += tree[i];
}
return res;
}
void update(int x){
for(int i = x; i <= len; i += lowbit(i)){
tree[i]++;
}
}
};
class Solution {
public:
int reversePairs(vector<int>& nums) {
int len = nums.size(), ans = 0;
if(!len) return ans;
vector<int> tmp = nums;
sort(tmp.begin(), tmp.end());
for(int &num : nums){
num = lower_bound(tmp.begin(), tmp.end(), num) - tmp.begin() + 1; // 离散化
}
BIT bit(len);
for(int i = len - 1; i >= 0; i--){
ans += bit.query(nums[i] - 1);
bit.update(nums[i]);
}
return ans;
}
};