插个题:在解决这个算法题之前,先看一下冒泡排序:
借个大佬的图:
对于冒泡,是相邻两个元素比较,大的数向后移,每循环一次就确定一个“最大的数”,排在最后,下一次比较时就不用再排了,所以代码的第二个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;
}