在两个有序序列中,找到两序列里第k大的元素log n时间

在算法书上学到的内容,大概看懂了重新试一下,留着以后看,如果能帮到大家最好,如果大家觉得我的注释很麻烦请勿喷,我是菜鸡。

#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;
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值