LintCode 532:Reverse Pairs

来源:互联网 发布:fzl什么意思网络用语 编辑:程序博客网 时间:2024/06/05 18:11

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 herelong long count = 0;SubReversePairs(A, 0, A.size() - 1, count);return count;}};