个人记录-LeetCode 23. Merge k Sorted Lists

最新推荐文章于 2021-01-29 10:44:49 发布

原创最新推荐文章于 2021-01-29 10:44:49 发布 · 642 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

108 篇文章

订阅专栏

LeetCode练习记录

108 篇文章

订阅专栏

这篇博客探讨了如何解决LeetCode中的23题，即合并多个已排序的链表。由于题目没有指定排序规则，作者提出使用空间换取时间的策略来简化问题。文章提供了相应的代码示例来展示如何实现这个方法。

问题：
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

问题要求：合并K个已经排好序的链表。
这个问题麻烦的地方在于：并没有定义链表排序的规则，即不知道升序还是降序。
同时链表的数量较多，判断和合并比较麻烦。

因此，最直观简便的做法，就是用空间换时间。

代码示例：

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null) {
            return null;
        }

        //定义一个TreeMap，key为链表的值，value为值出现的次数
        Map<Integer, Integer> valueWithNum = new TreeMap<>();

        //轮询所有的链表，将值的信息加入到TreeMap
        for (ListNode head : lists) {
            while (head != null) {
                if (valueWithNum.containsKey(head.val)) {
                    valueWithNum.put(head.val, valueWithNum.get(head.val) + 1);
                } else {
                    valueWithNum.put(head.val, 1);
                }
                head = head.next;
            }
        }

        ListNode temp = new ListNode(0);
        ListNode result = temp;
        //从TreeMap中重新构造出新的链表
        for (Integer key : valueWithNum.keySet()) {
            int num = valueWithNum.get(key);
            while (num > 0) {
                --num;
                temp.next = new ListNode(key);
                temp = temp.next;
            }
        }

        return result.next;
    }
}