描述
在数组中的两个数字如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。给你一个数组,求出这个数组中逆序对的总数。
概括:如果a[i] > a[j] 且 i < j, a[i] 和 a[j] 构成一个逆序对。
您在真实的面试中是否遇到过这个题? 是
样例
序列[2, 4, 1, 3, 5] 中,有 3 个逆序对(2, 1), (4, 1), (4, 3),则返回 3 。
在数组中的两个数字如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。给你一个数组,求出这个数组中逆序对的总数。
概括:如果a[i] > a[j] 且 i < j, a[i] 和 a[j] 构成一个逆序对。
您在真实的面试中是否遇到过这个题? 是
样例
序列[2, 4, 1, 3, 5] 中,有 3 个逆序对(2, 1), (4, 1), (4, 3),则返回 3 。
程序
class Solution {
public:
/**
* @param A: an array
* @return: total of reverse pairs
*/
/*
解题思路:如果按照一般的做法,利用两个for循环,那么时间复杂度就是O(n*n);利用归并排序的思想时间复杂度是O(nlogn)。两个指针分别指向前半部分的尾部 ,和后半部分的尾部,进行比较,如果前半部分尾部的数字大于后半部分尾部的数字前半部分尾部的数字一定比后半部分的都大,那么逆序列数字累加后半部分数字个数,在讲前半部分尾部的数字copy到拷贝数组中,防止重复比较,在比较后将copy数组复制到data。
*/
long long reversePairs(vector<int> &A) {
// write your code here
long long length = A.size();
return mergeSort(A, 0, length - 1);
}
long long mergeSort(vector<int> &data, long long start, long long end){
if(start >= end) return 0;
long long mid = (start + end) / 2;
long long left = mergeSort(data, start, mid);
long long right = mergeSort(data, mid + 1, end);
long long ForIdx = mid;
long long BackIdx = end;
long long copyIdx = end;
long long count = 0;
vector<int> copy(data);
while(ForIdx >= start && BackIdx >= mid + 1){
if(data[ForIdx] > data[BackIdx]){
copy[copyIdx--] = data[ForIdx--];
count += BackIdx - mid;
}
else
copy[copyIdx--] = data[BackIdx--];
}
while(ForIdx >= start)
copy[copyIdx--] = data[ForIdx--];
while(BackIdx >= mid + 1)
copy[copyIdx--] = data[BackIdx--];
for(int i = start; i <= end; i++)
data[i] = copy[i];
return left + right + count;
}
};