数组-两个排序数组的中位数-困难

描述
两个排序的数组A和B分别含有m和n个数，找到两个排序数组的中位数，要求时间复杂度应为O(log (m+n))。

样例
给出数组A = [1,2,3,4,5,6] B = [2,3,4,5]，中位数3.5
给出数组A = [1,2,3] B = [4,5]，中位数 3

挑战

时间复杂度为O(log n)

题目链接

分析

因为时间复杂度为O(log n)，所以可以使用归并排序，将A和B共同排序之后的结果保存在新的数组中。判断新数组的长度是否为奇或者为偶，如果是奇数，返回中间的那个数；如果是偶数，将中间两端的数相见求和。这里返回的数是一个double类型，所以必要的时候需要强制转换。

程序

class Solution {
public:
	/*
	* @param A: An integer array
	* @param B: An integer array
	* @return: a double whose format is *.5 or *.0
	*/
	double findMedianSortedArrays(vector<int> &A, vector<int> &B) {
		// write your code here
		vector<int> result;
		int i = 0;
		int j = 0;
		while (i < A.size() && j < B.size())//归并排序
		{
			if (A[i] <= B[j])
				result.push_back(A[i++]);
			else
				result.push_back(B[j++]);
		}
		while (i < A.size()){
			result.push_back(A[i++]);
		}
		while (j < B.size()){
			result.push_back(B[j++]);
		}
		for (int i = 0; i < result.size(); i++)
			cout << result[i] << " ";
		cout << endl;
		if (result.size() % 2 == 0){//偶数
			int result1 = result[result.size() / 2 - 1];
			int result2 = result[result.size() / 2];
			double result = (result1 + result2) / 2.0;
			return result;
		}
		else//奇数
		{
			return double(result[result.size() / 2]);
		}
	}
};