算法实验1-4查找两个有序数组合并之后的中位数

插个题：在解决这个算法题之前，先看一下冒泡排序：
借个大佬的图：

对于冒泡，是相邻两个元素比较，大的数向后移，每循环一次就确定一个“最大的数”，排在最后，下一次比较时就不用再排了，所以代码的第二个for循环是n-i-1。

---

步入正题
问题：

设两个长度分别为m和n的有序数组a[0:m-1]和b[0:n-1]，请使用分治策略找出这两个有序数组合并之后的中位数

算法思路：由于两个数组合并之后的中位数，比较nums1和nums2的中位数a,b，若a<b，则合并数组的中位数在a的右边，b的左边。若a>b，则合并数组的中位数在a的左边，b的右边。此题采用的是二分查找类似，此题对应的二分的终止条件是：是某个数组小到无法继续二分的时候。这是因为，每次二分剔除掉的是更少的那个部分。因此，在终止条件中，查找范围应该是一个大数组和一个只有 1～2 个元素的小数组。这样就需要根据大数组的奇偶性和小数组的数量。
终止条件 分为了5个可能性：（1）如果 a, b都只剩一个元素，返回其中的最小值即可。（2）如果a，b都剩两个元素，比较a首位值和b尾部值的大小。（3）3）当b的个数为1时 a的个数为偶数时。（4）当b的个数为1时， a的个数为奇数时。（5）（5）b的个数为2时，a的个数为奇数或偶数的判断情况是一样的。具体的可以查看代码里的详细注释！！！

上代码：

#include<iostream>
using namespace std;
#define min(a,b) (a>b? b:a)
/*
	if(n>=m)
		cout<<"两组有序数列的合并数列中位数为:"<<find(a,0,n - 1,b,0,m - 1);
	if(n<m)
		cout<<"两组有序数列的合并数列中位数为:"<<find(b,0,m - 1,a,0,n - 1);
*/
//参数分别代表数组a，数组a的首部，数组a的尾部
//数组b，数组b的首部，数组b的尾部
//准确来说数组a的数组长度更大，长度大的在前面 
int find(int a[],int begina,int enda,int b[],int beginb,int endb){
	//数组a、b的长度 
	int lengtha = enda - begina + 1;
	int lengthb = endb - beginb + 1;
	
	//mid为数组a的中位数的索引 
	int mid = (enda + begina) / 2;
	//（1）如果 a, b都只剩一个元素，返回其中的最小值即可 
	if(lengtha == 1&&lengthb == 1 ){
		return min(a[begina],b[beginb]);
	} 
	
	//（2）如果a，b都剩两个元素，比较a首位值和b尾部值的大小 
	if(lengtha == 2 &&lengthb == 2){
	//如果a的首部小，选a尾部和b首部中最小的 
		if(a[begina] < b[beginb]){
			return min(a[enda],b[beginb]);
		}
		//反之 
		else{
			return min(a[begina],b[endb]);
		}
	} 
	
	//假定a的元素个数大于b的元素个数
	//（3）当b的个数为1时 a的个数为偶数时 
	if((lengtha % 2 == 0 )&& (lengthb == 1)){ 
		int m = b[beginb];
		int c = a[mid];
		int d = a[mid + 1];
		//比较m、c、d三个值，看谁位于中间值，谁就是中位数
		/*
		如果 m < c，则结果为 c；
		果 c < m < d，则结果为 m；
		如果m > d，则结果为 d。
		*/ 
		return  m < c ? c : (m < d ? m : d);
	}
	
	//（4）当b的个数为1时， a的个数为奇数时 
	if((lengtha % 2 == 1) && (lengthb == 1)){
		
		//比较m、c、bb三个值，看谁位于中间值，谁就是中位数
		/*如果 m < bb，则结果为 bb；
		如果 bb< m < c，则结果为 m；
		如果 m > c，则结果为 c。
		*/ 
		int bb = a[mid - 1];
		int c = a[mid];
		int m = b[beginb];
		return m < bb ? bb : (m < c ? m : c);
	
	}
	
	//当（5）b的个数为2时，a的个数为奇数或偶数的判断情况是一样的 
	/*
	可能性三：nums1 奇数个，nums2 有 2 个元素。例如，nums1 = [a, b, c, d, e]，nums2 = [m,n]。
	此时，有以下 6 种可能性：
	如果 n < b，则结果为 b；
	如果 b < n < c，则结果为 n；
	如果 c < n < d，则结果为 max(c,m)；
	如果 n > d，m < c，则结果为 c；
	如果 n > d，c < m < d，则结果为 m；
	如果 n > d，m > d，则结果为 d。
	其中，4～6 可以合并为，如果 n > d，则返回 m < c ? c : (m < d ? m : d)。
	可能性四：nums1 偶数个，nums2 有 2 个元素。结果与上述一样*/
	if(lengthb == 2){
		int bb = a[mid - 1];
		int c = a[mid];
		int d = a[mid + 1];
		int m = b[beginb];
		int n = b[endb];
		if(n < bb){
			return bb;
		}else if(n < c){
			return n;
		}else if(n < d){
			return c > m ? c : m;
		}else{
			return m < c ? c : (m < d ? m : d);
		}
	}	
	//划分
	int mida = (begina + enda) / 2;
	int midb = (beginb + endb) / 2;
	/*划分原则 :
	中位数比一组中间数大的小（在左边）， 
	比一组中间数小的大（在右边），
	两组增删一致，按b的即元素少的那一组 */ 
	if( a[mida] < b[midb]){
		int steps = endb - midb;
		return find(a,begina + steps,enda,b,beginb,endb - steps);
	} else {
		int steps = midb - beginb;
		return find(a,begina,enda - steps,b,beginb + steps,endb);
	}
}
int main(){
	int n,m;
	cout<<"输入两个数组的元素个数："; 
	cin>>n;
	cin>>m;
	int a[n],b[n];
	cout<<"输入"<<n<<"个元素：";
	for(int i=0;i<n;i++)
		cin>>a[i]; 
	cout<<"输入"<<m<<"个元素：";
	for(int i=0;i<m;i++)
		cin>>b[i];

	if(n>=m) //先比较两个序列的长度 ，长的放在最前面 
		cout<<"两组有序数列的合并数列中位数为:"<<find(a,0,n - 1,b,0,m - 1);
	if(n<m)
		cout<<"两组有序数列的合并数列中位数为:"<<find(b,0,m - 1,a,0,n - 1);
	return 0;
}