每日一题——合并K个排序链表

菜鸡每日一题系列打卡23天

每天一道算法题目 

小伙伴们一起留言打卡

坚持就是胜利，我们一起努力！

题目描述(引自LeetCode)

合并k个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:
输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

题目分析

本题是合并两个有序链表的进阶版本，之前做过合并两个有序链表的题目的小伙伴应该首先可以想到顺序合并的解决方案，依次合并每两个链表，最终合并为一个有序的链表。进一步考虑，可以采用分治的思想，每次合并一半的链表，最终得到一个有序的链表。由于顺序合并的方式仅仅是在之前合并两个有序链表的题目上加一层循环，而且时间复杂度较分治的实现方式更高，因此，本文将采用分治的方式进行实现。

代码实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {


   public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) return null;
        return merge(lists, 0, lists.length - 1);
    }


    private ListNode merge(ListNode[] lists, int start, int end) {
        if (start == end) return lists[start];
        int mid = start + ((end - start) >> 1);
        return mergeTwoLists(merge(lists, start, mid), merge(lists, mid + 1, end));
    }


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

代码分析

对代码进行分析，设有k个链表，所有链表的元素总和为n，采用分治的方式进行实现，则其代码的时间复杂度为O(kn * logk)，而就空间而言，所占用的空间与递归深度有关，因此，空间复杂度为O(logk)。

执行结果

学习 | 工作 | 分享

????长按关注“有理想的菜鸡”

只有你想不到，没有你学不到