欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2024-03-15 22:00:48
...

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