[LeetCode]合併K個排序鏈表(Merge k Sorted Lists)

題目描述

合併 k 個排序鏈表,返回合併後的排序鏈表。請分析和描述算法的複雜度。java

示例:算法

輸入:
[
1->4->5,
1->3->4,
2->6
]
輸出: 1->1->2->3->4->4->5->6

ListNode數據結構數據結構

class ListNode {
    int val;
    ListNode next;

    ListNode(int x) {
        val = x;
    }
}

解決方法

相鄰鏈表兩兩合併,兩兩合併詳情見合併兩個有序鏈表
spa

public ListNode mergeKLists(ListNode[] lists) {
        int interval = 1;

        while (interval < lists.length) {
            for (int i = 0; i < lists.length - interval; i += interval * 2) {
                lists[i] = merge2Lists(lists[i], lists[i + interval]);
            }
            interval *= 2;
        }

        return lists.length > 0 ? lists[0] : null;
    }

    public ListNode merge2Lists(ListNode l1, ListNode l2) {
        if (l1 == null)
            return l2;
        if (l2 == null)
            return l1;
        if (l1.val < l2.val) {
            l1.next = merge2Lists(l1.next, l2);
            return l1;
        } else {
            l2.next = merge2Lists(l1, l2.next);
            return l2;
        }
    }

時間複雜度: O(Nlogk),k是鏈表的數目code

本文首發:https://lierabbit.cn/2018/09/...blog

相關文章
相關標籤/搜索