LeetCode 4. 寻找两个有序数组的中位数

题目：

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3] nums2 = [2] 则中位数是 2.0

示例 2:

nums1 = [1, 2] nums2 = [3, 4] 则中位数是 (2 + 3)/2 = 2.5

思路：构建一个最大堆一个最小堆，最大堆放较小的元素，最小堆放较大的元素，并且尽量保证二者存储的数的个数一样多，这样二者堆顶和的平均值，或者某一个较大堆的堆顶就是中位数。

#include<vector>
#include<queue>
#include<functional>//无此头文件greater非法
#include<iostream>
#include<cstdlib>

using namespace std;

class Solution {
public:
	priority_queue<int, vector<int>, greater<int> > minHeap;//最小堆，用来放较大的元素
	priority_queue<int, vector<int>, less<int> > maxHeap;//最大堆，用来放较小的元素
	double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
		double res = 0.0;
		for (int i = 0; i < nums1.size(); i++) {
			addNums(nums1[i]);
		}
		for (int i = 0; i < nums2.size(); i++) {
			addNums(nums2[i]);
		}
		if (maxHeap.size() == minHeap.size()) {
			res = (maxHeap.top()*1.0 + minHeap.top()*1.0) / 2.0;
		}
		else if (maxHeap.size() > minHeap.size())
			res = maxHeap.top()*1.0;
		else
			res = minHeap.top()*1.0;
		return res;
	}
	//往两个队里添加数字
	void addNums(int num) {
		//如果最大堆
		if (maxHeap.empty()) {
			maxHeap.push(num);
			return;//要返回不然第一个数会重复加入堆中
		}
		//如果两个堆的数一样多就直接将num和两个堆顶比较，将大的放在最小堆，小的放在最大堆
		if (maxHeap.size() == minHeap.size()) {
			if (num > minHeap.top())
				minHeap.push(num);
			else
				maxHeap.push(num);
		}
		//如果最大堆的数比最小堆的多
		else if (maxHeap.size() > minHeap.size()) {
			//如果num大于最大堆堆顶直接push进最小堆
			//（这里不能和最小堆堆顶比较，否则不能保证最小堆的数都比最大堆的大）
			if (num > maxHeap.top())
				minHeap.push(num);
			//否则将最大堆堆顶push进最小堆，再将num放进最大堆
			else{
				minHeap.push(maxHeap.top());
				maxHeap.pop();
				maxHeap.push(num);
			}
		}
		else if (maxHeap.size() < minHeap.size()) {
			//如果num小于最小堆堆顶直接push进最大堆
			if (num < minHeap.top())
				maxHeap.push(num);
			//否则将最小堆堆顶push进最大堆，再将num放进最小堆
			else {
				maxHeap.push(minHeap.top());
				minHeap.pop();
				minHeap.push(num);
			}
		}
	}
};

int main() {
	vector<int> nums1;
	vector<int> nums2;
	nums1.push_back(1);
	nums1.push_back(3);
	nums2.push_back(2);
	Solution solve;
	double res= solve.findMedianSortedArrays(nums1,nums2);
	cout << res << endl;
	system("pause");
	return 0;
}