LintCode 532:Reverse Pairs

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;
	}
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值