23. Merge k Sorted Lists

题干

Merge k Sorted Lists
Difficulty: Hard
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

合并K个已经排序的数组。

解题思路:

难度感觉算不上Hard(毕竟我做不出来的都是Hard),不过为了方便调试这次用了.net来写。
拿到题目最开始的反应是对于数组中的每一个链表都定义一个指针,每次遍历所有的指针,记录最小的值以及出现的位置,遍历完成后将最小值插入结果链表中,如果遍历后发现每一个指针都指向空节点则说明遍历完成。
最后成品:

        public ListNode MergeKLists(ListNode[] lists)
        {
            ListNode head = null, cursor = null;
            while (true)
            {
                var indexs = new List();
                var min = int.MaxValue;
                for (var i = 0; i < lists.Length; i++)
                {
                    if (lists[i] == null)
                    {
                        continue;
                    }
                    if (lists[i].val > min)
                    {
                        continue;
                    }
                    else
                    {
                        if (lists[i].val < min)
                        {
                            indexs.Clear();
                            min = lists[i].val;
                        }
                        indexs.Add(i);
                    }
                }
                if (min == int.MaxValue)
                {
                    break;
                }
                else
                {
                    for (var i = 0; i < indexs.Count; i++)
                    {
                        if (head == null)
                        {
                            head = new ListNode(min);
                            cursor = head;
                        }
                        else
                        {
                            cursor.next = new ListNode(min);
                            cursor = cursor.next;
                        }
                        lists[indexs[i]] = lists[indexs[i]].next;
                    }
                }
            }
            return head;
        }

Runtime: 112 ms 91.21%
时间复杂度O(K*log(N))
K是链表数组的长度,N是节点数量。
Memory Usage: 28.8 MB 8.33%
空间复杂度O(K+N)
其中O(N)指的是最后长度为N的排序后数组。
O(K)则是我为了存储指针位置而新建的List。
时间上我很满意了,但是空间复杂度看上去很高,试着去减少一下。
结果链表的空间肯定是不能省的,那么只能省去List了,每次只push一个最小值进去,只记录最小值以及最小值的位置

        public ListNode MergeKLists(ListNode[] lists)
        {
            ListNode head = null, cursor = null;
            while (true)
            {
                var index = -1;
                var min = int.MaxValue;
                for (var i = 0; i < lists.Length; i++)
                {
                    if (lists[i] == null)
                    {
                        continue;
                    }
                    if (lists[i].val >= min)
                    {
                        continue;
                    }
                    else
                    {
                        min = lists[i].val;
                        index = i;
                    }
                }
                if (index == -1)
                {
                    break;
                }
                else
                {

                    if (head == null)
                    {
                        head = new ListNode(min);
                        cursor = head;
                    }
                    else
                    {
                        cursor.next = new ListNode(min);
                        cursor = cursor.next;
                    }
                    lists[index] = lists[index].next;
                }
            }

            return head;
        }

Runtime: 584 ms
Memory Usage: 29.3 MB
这样做肯定增加时间的,时间复杂度应该为(O(KN)),但是空间复杂度应该能缩减为O(N)。
但是结果让我大跌眼镜,空间占用不减反增,这里是实在不能理解。

其余解法

回去翻看LeetCode上的solution,他列出了五种解法。
最简单的应该就是暴力求解了,将所有的链表中的节点都push到一个数组中,排序后生成新的链表。
时间O(N*log(N)),空间O(N)。

另一个Smart Play则是将问题简化为(K-1)次合并两个排序数组,我个人觉得思路比我的好,很多复杂问题都能简化为多个简单问题的累加。时间复杂度O(KN),空间复杂度O(1)。

最后一种解法则是上一个方法的改进,每次合并floor(k/2)个数组,这样合并次数就被减少到ceil(log2(k))次了。这样就减少了数组中大部分链表的结点需要排序的次数,而后几个链表的元素排序次数则会略有增加,大部分情况下优于上面的解。除非最后一个链表节点特别多,其余链表节点特别少,这样的话上一个做法会优于这个。

你可能感兴趣的:(23. Merge k Sorted Lists)