如何将两个排序的数组合并为一个排序的数组? [关闭]

问题

我在接受采访时被问及这是我提供的解决方案:

public static int[] merge(int[] a, int[] b) {

    int[] answer = new int[a.length + b.length];
    int i = 0, j = 0, k = 0;
    while (i < a.length && j < b.length)
    {
        if (a[i] < b[j])
        {
            answer[k] = a[i];
            i++;
        }
        else
        {
            answer[k] = b[j];
            j++;
        }
        k++;
    }

    while (i < a.length)
    {
        answer[k] = a[i];
        i++;
        k++;
    }

    while (j < b.length)
    {
        answer[k] = b[j];
        j++;
        k++;
    }

    return answer;
}

有没有更有效的方法来做到这一点?

编辑:更正长度方法。


#1 热门回答(97 赞)

public static int[] merge(int[] a, int[] b) {

    int[] answer = new int[a.length + b.length];
    int i = 0, j = 0, k = 0;

    while (i < a.length && j < b.length)  
       answer[k++] = a[i] < b[j] ? a[i++] :  b[j++];

    while (i < a.length)  
        answer[k++] = a[i++];


    while (j < b.length)    
        answer[k++] = b[j++];

    return answer;
}

有点紧凑但完全一样!


#2 热门回答(51 赞)

我很惊讶没有人提到这个更酷,更高效和紧凑的实现:

public static int[] merge(int[] a, int[] b) {
    int[] answer = new int[a.length + b.length];
    int i = a.length - 1, j = b.length - 1, k = answer.length;

    while (k > 0)
        answer[--k] =
                (j < 0 || (i >= 0 && a[i] >= b[j])) ? a[i--] : b[j--];
    return answer;
}

兴趣点-请注意,它与其他任何O(n)算法的操作数相同或更少,但在单个while循环中只有单个语句!

  • 如果两个数组的大小大致相同,则O(n)的常数相同。但是,如果数组真的不平衡,那么使用System.arraycopy的版本将获胜,因为在内部它可以使用单个x86汇编指令执行此操作。
  • 注意a [i]> = b [j]而不是[i]> b [j]。这保证了"稳定性",定义为当a和b的元素相等时,我们想要来自a之前的元素。

#3 热门回答(31 赞)

一个小的改进,但在主循环之后,你可以使用System.arraycopy来复制任何一个输入数组的尾部,当你到达另一个结束时。但是,这不会改变解决方案的O(n)性能特征。