在两个升序排列的数组中找到第K大的元素(分治法,C/C++)

给出已经排序数组,a和b,长度分别为n、m,请找出一个时间复杂度为O(log2(n+m))的算法,找到排在第k位置的元素。

#include <stdio.h>
int findk(int a[], int n, int b[], int m, int k);

int main(){
    int i,result;
    int a[]={1,5,8};
    int b[]={2,3,4,6,7};
    int n = sizeof(a)/sizeof(a[0]);
    int m = sizeof (b)/sizeof(b[0]);


    for(i=1; i <= n + m; i++){
        result =findk(a,n,b,m,i);
        printf("第%d小的元素是:%d\n",i,result);
    }

}

//在两个升序排列的数组中找到第K大的元素
int findk(int a[], int n, int b[], int m, int k) {
    if(k<=0)return -1;
    if(n>m)                                 //用于保证n<=m,即保证前一个数组的元素较少
        return findk(b,m,a,n,k);
    if(n==0)
        return b[k-1];
    if(k==1)
        return a[0]<b[0]?a[0]:b[0];
    int numa=(n>=k/2)?k/2:n;                //当数组中没有k/2个元素时取n
    int numb=k-numa;
    if(a[numa-1]==b[numb-1])
        return a[numa-1];
    else if(a[numa-1]>b[numb-1])
        return findk(a,n,&b[numb],m-numb,k-numb);
    else
        return findk(&a[numa],n-numa,b,m,k-numa);


}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值