在算法书上学到的内容,大概看懂了重新试一下,留着以后看,如果能帮到大家最好,如果大家觉得我的注释很麻烦请勿喷,我是菜鸡。
#include<iostream>
using namespace std;
/* 两有序数组求其中第k大的数 */
int find_kth(int *a,int n,int *b, int m,int k){ //函数主体
if(k < 0){
return -1;
}
if(m < n){ //因为默认前面那个数组长度更小,所以如果初始传入前面的大,就重新调用一次该函数
find_kth(b,m,a,n,k);
}
if(n == 0){ //考虑到前面数组长度为0的情况,就直接从b数组找到第k个元素返回即可
return b[k - 1];
}
if(k == 1){ //又考虑到k = 1的情况,也就是停止条件的一部分,直接选a,b两数组第一个元素较小的那个即可
return (a[0] <= b[0]) ? a[0] : b[0];
}
int numa = min(k/2,n); //该步骤开始主要参考二分查找,不是从头开始找,而是从中部开始查找,省去挨个查找时间
int numb = k - numa; //numa + numb = k,可以让两个数组当中的两个位置前面的元素加起来刚好是k
if(a[numa - 1] == b[numb - 1]){
return a[numa - 1]; //如果第numa和第numb个元素相等,直接返回其中一个即可,因为下标numa前面有numa个元素,下标numb前面有numb个元素,加起来刚好是第k大的数
}
else if(a[numa - 1] > b[numb - 1]){ //如果前者大,那说明第k大的元素暂时不在a里,应该去b数组的后部分去找,所以递归调用函数,a原封不动传入,而b只传入numb之后的,要找也只找第k - numb大的元素
return find_kth(a,n,b+numb,m - numb, k - numb);
}
else if(a[numa - 1] < b[numb - 1]){
return find_kth(a + numa,n - numa,b,m, k - numa);
}
}
int main(){
int n,m,k;
cout << "请分别输入a数组长度,b数组长度,以及要查询的第k大的数,用空格分隔:";
cin >> n >> m >> k;
int *a = new int[n];
int *b = new int[m];
for(int i = 0; i < n ; i++){
cin >> a[i];
}
for(int i = 0; i < m ; i++){
cin >> b[i];
}
int result = find_kth(a,n,b,m,k);
cout << "第" << k << "大的数是:"<<result << endl;
return 0;
}