Question

我写了一个合并K个排序数组。我发现在其他站点上，最佳时间复杂度为O（n k Logk），其中k是数组的数目，n是每个数组中的元素的数目。我认为我是O（n k）。

有人可以确认吗？代码如下。

private static void MergeKSortedArrays()
{
    int[][] arr = { new int[] { 3, 5, 7 }, new int[] { 1, 2, 4 }, new int[] { 6, 8, 9 } };
    int k = 3, n = 3;

    int[] output = new int[n * k];
    int[] temp = new int[k];

    for (int i = 0; i < k - 1; i++)
    {
        temp = Merge(arr[i], arr[i + 1]);  // takes Linear time
        arr[i + 1] = temp;
    }

    foreach(int i in arr[k-1])
    {
        Console.Write(i + " ");
    }
    Console.WriteLine();

}

private static int[] Merge(int[] a, int[] b)
{
    int[] o = new int[a.Length + b.Length];
    int i = 0, j = 0, ind = 0;

    for (; i < a.Length && j < b.Length;)
    {
        if (a[i] <= b[j])
        {
            o[ind] = a[i];
            i++;
            ind++;
        }
        else
        {
            o[ind] = b[j];
            j++;
            ind++;
        }
    }

    if (i < a.Length)
    {
        for (; i < a.Length; i++, ind++)
        {
            o[ind] = a[i];
        }
    }
    else if (j < b.Length)
    {
        for (; j < b.Length; j++, ind++)
        {
            o[ind] = b[j];
        }
    }

    return o;
}

Answer 1

否。

在第一次迭代中，将长度为n的数组与长度为n的数组合并

在第二次迭代中，将长度为n的数组与长度为2n的数组合并

在第三次迭代中，您将长度为n的数组与长度为3n的数组合并

...

这意味着Merge（）方法中的for循环将运行2n + 3n + 4n ... = (k+1)*k/2 * n -1次。

所以您建议的算法实际上是O(n * k^2)

确定合并K个排序数组的时间复杂度

1 个答案: