Description:
For an array A, if i < j, and A [i] > A [j], called (A [i], A [j]) is a reverse pair.
return total of reverse pairs in A.
Note:
1.采用分治思想(类似于mergesort),先计算一个subarray中的reverse pairs,然后再计算两个subarray merge后产生的reverse pairs,为了不重复进行计算,需要计算完一个subarray的结果后对这其进行排序,排序后,两个subarray的前后相对关系还是不影响,所以内部进行的排序不影响后续merge的结果。
2.两个排完序的subarrray进行merge时,就可以看到有些比较是多余的、可以免去的。当后一个数组(初始位置为b0)的某个数bn小于前一个数组的某个数a时,在bn之前的数字也必定小于a,则产生n对reverse pair.
3.就跟mergesort一样,我们需要O(N)的辅助空间。我们可以提前生成好数组,也可以用到时再分配(效率会低一点,但表面上看上去直观一点),注意分配完动态数组后及时析构。
Code:
class Solution {
void Merge(vector<int>& A, int begin, int mid, int end, long long& count) {
int first = mid;
int second = end;
int* tmp = new int[end - begin + 1];
int i = end - begin;
while (first > begin-1 && second > mid) {
if (A[second] < A[first]) {
count += second - mid;
tmp[i] = A[first];
first--;
}
else {
tmp[i] = A[second];
second--;
}
i--;
}
while (first > begin - 1) {
tmp[i] = A[first];
i--;
first--;
}
while (second > mid) {
tmp[i] = A[second];
i--;
second--;
}
for (int i = 0; i < end - begin + 1; i++) {
A[begin + i] = tmp[i];
}
delete tmp;
}
void SubReversePairs(vector<int>& A, int begin, int end, long long& count) {
if (begin >= end)
return;
int mid = begin + (end - begin) / 2;
SubReversePairs(A, begin, mid, count);
SubReversePairs(A, mid + 1, end, count);
Merge(A, begin, mid, end, count);
}
public:
/**
* @param A an array
* @return total of reverse pairs
*/
long long reversePairs(vector<int>& A) {
// Write your code here
long long count = 0;
SubReversePairs(A, 0, A.size() - 1, count);
return count;
}
};