Question

我试图从两个排序的数组中获取最大的K元素而不合并它们。我的代码使用分治法，我试图将两个数组与中间元素分开，然后检查中间一个数组是否大于另一个数组，然后将数组的末尾移到中间，减小k到中间，我将另一个数组从它的开始移到中间。

我的问题是输入大，导致堆栈溢出异常。我不知道为什么，但是我认为这是因为如果中间的减数和数组的长度大于K且我无法解决，则该语句将数组的起始位置移到中间。

谁能帮我一个忙，请帮我吗？

public static int GetKthItem(int[] arr1, int[] arr2, int N, int M, int K, 
int start1, int start2, int end1, int end2){
if (end1 + end2 == K)
{
    return Math.Max(arr1[0], arr2[0]);
}

if (end1 + end2 + 1 == K)
{
    return Math.Min(arr1[0], arr2[0]);
}

if (start1 == end1 && K > 0)
{
    K--;
    if (K > end2 - start2)
        return arr2[start2];
    else
        return arr2[end2 - K];
}
else if (start2 == end2 && K > 0)
{
    K--;

    if (K > end1 - start1)
        return arr1[start1];
    else
        return arr1[end1 - K];
}

if (K == 0)
{
    return Math.Max(arr1[end1], arr2[end2]);
}

int mid1 = (end1 - start1) / 2;
int mid2 = (end2 - start2) / 2;

if (arr2[mid2] >= arr1[mid1])
{
    if (end2 - mid2 <= K)
    {
        return GetKthItem(arr1, arr2, N, M, (K - (end2 - mid2)), mid1, 
            start2, end1, mid2);
    {
    else
    {
        return GetKthItem(arr1, arr2, N, M, K, mid1, start2, end1, end2);
    }
}
else
{
    if (end1 - mid1 <= K)
    {
        return GetKthItem(arr1, arr2, N, M, (K - (end1 - mid1)), start1, 
            mid2, mid1, end2);
    {
    else
    {
        return GetKthItem(arr1, arr2, N, M, K, start1, mid2, end1, end2);
    }
}
}

Answer 1

好吧，如果两个数组已经排序，则需要获得最大的K个元素。您可以有2个指针P1和P2。 P1 =数组1的最后一个元素[已经排序的将是最后一个元素] P2 =数组2的最后一个元素

然后

result[] -- new result array
index = 0 
while(index<k){
   if (Arr1[P1] > Arr2[P2]){
     result[index] = Arr1[P1]
     P1--
   } else {
     result[index] = Arr2[P2]
     P2--
   }
   index++
}

从两个排序数组中获取前K个项目，而不将它们组合

1 个答案: