两个有序数组中找到第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];
}
}
推荐阅读
-
两个有序数组中找到第k大的元素
-
在两个排序数组中找第k小的数
-
给定一个整数,在一个整数数组中找到两个下标不同的元素,其和为给定的整数
-
《剑指offer》-递增数组中找到和为S的(最小)两个元素
-
寻找数组中第K大的数,时间复杂度O(N)
-
找出数组中第k大的数(时间复杂度分析、C++代码实现). TopK in array. ( leetcode - 215 )
-
不能用排序、求一个数组第k大的数字、时间复杂度为O(nlog₂k)
-
找出数组中第K大的数,时间复杂度O(n)
-
LeetCode 378有序矩阵中第K小的元素
-
[Leedcode][第215题][JAVA][数组中的第K个最大元素][快排][优先队列]