在数组中的两个数字如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。
输入一个数组,求出这个数组中的逆序对的总数。
样例
输入:[1,2,3,4,5,6,0]
输出:6
双重循环O(N^2)写法:
class Solution {
public:
int inversePairs(vector<int>& nums) {
int res = 0;
for(int i = 0; i < nums.size(); i++)
for(int j = i + 1; j < nums.size(); j++)
if(nums[i] > nums[j]) res++;
return res;
}
};
归并排序O(nlgn)
C++代码:
class Solution {
public:
int merge(vector<int> &nums, int l,int r){
if(l >= r) return 0;
int mid = l + r >> 1;
int res = merge(nums,l,mid) + merge(nums,mid+1,r);
vector<int> temp;
int i = l, j = mid + 1;
while(i <= mid && j <= r){
if(nums[i] <= nums[j]) temp.push_back(nums[i++]);
else{
temp.push_back(nums[j++]);
res += mid - i + 1;
}
}
while(i <= mid) temp.push_back(nums[i++]);
while(j <= r) temp.push_back(nums[j++]);
int k = l;
for(auto x : temp) nums[k++] = x;
return res;
}
int inversePairs(vector<int>& nums) {
return merge(nums,0,nums.size() - 1);
}
};