两个有序数组中找到第k大的元素

1. 求第k大的数,可以定义两个游标(i,j)分别指向两个有序数组(a[], b[]),a[i] > b[j] 则j++,result = b[j];否则i++。移动一次则count++,当count等于k时,返回result。

代码如下:

#include <iostream>

int getRankData(int a[], int aSize, int b[], int bSize, int rank)
{
    int i = 0;
    int j = 0;
    if (aSize + bSize < rank)
    {
        return -1;
    }
    int totalRank = 0;
    int result(0);
    for (; i < aSize && j < bSize && totalRank < rank;)
    {
        totalRank++;
        std::cout << "a:" << a[i] << ", b:" << b[j] << std::endl;
        if (a[i] > b[j])
        {
            result = b[j++];
        }
        else
        {
            result = a[i++];
        }
    }
    if (totalRank == rank)
    {
        return result;
    }
    if (i == aSize)
    {
        result = b[rank - totalRank  - 1 + j];
    }
    else
    {
        std::cout << "totalRank:" << totalRank << ", i:" << i << std::endl;
        result = a[rank - totalRank - 1 + i];
    }
    return result;
}

int main()
{
    int a[] = {1, 3, 4, 6, 8, 9};
    int b[] = {4, 6, 7};
    std::cout << "result:" << getRankData(a, sizeof(a)/ sizeof(int), b, sizeof(b) / sizeof(int), 8) << std::endl;
}

2.类二分查找

i + j = k, a[i] > b[j], 则排序为K的数不在b[0] 和 b[j] 之间,过滤掉j个元素,下次查找排名为rank = k - j;否则不在a[0]和a[i]之间,过滤掉i个元素,下次查找排名为rank = k - i。

结束条件:

1.其中一个数组为空,则返回a[rank - 1] 或 b[rank - 1];

2.rank = 1,则a[0] > b[0], 返回b[0],否则返回a[0];

int min(int para1, int para2)
{
    return para1 > para2 ? para2 : para1;
}

int getRankData(int a[], int aSize, int b[], int bSize, int rank)
{
    if ((aSize == 0 && bSize == 0) || (0 == rank))
    {
        return -1;
    }
    if (aSize > bSize)
    {
        return getRankData(b, bSize, a, aSize, rank);
    }
    if (aSize + bSize < rank)
    {
        return -1;
    }
    if (aSize == 0)
    {
        return b[rank - 1];
    }
    if (aSize + bSize == rank)
    {
        return min(a[aSize - 1], b[bSize - 1]);
    }
    if (rank == 1)
    {
        return min(a[0], b[0]);
    }

    int i = min(rank / 2, aSize);
    int j = rank - i;
    if (a[i - 1] > b[j - 1])
    {
        return getRankData(a, aSize, b + j, bSize - j, rank - j);
    }
    else if (a[i - 1] < b[j - 1])
    {
        return getRankData(a + i, aSize - i, b, bSize, rank - i);
    }
    else
    {
        return a[i - 1];
    }
}

 

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值