JAVA学习-练习试用Java实现“合并K个升序链表”

问题：

给定一个链表数组，每个链表都已经按升序排列。

请将所有链表合并到一个升序链表中，返回合并后的链表。

示例 1：

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：[  1->4->5,  1->3->4,  2->6]将它们合并到一个有序链表中得到。1->1->2->3->4->4->5->6
示例 2：

输入：lists = []
输出：[]
示例 3：

输入：lists = [[]]
输出：[]
提示：

k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i] 按 升序 排列
lists[i].length 的总和不超过 10^4
以下程序实现了这一功能，请你填补空白处内容：

public class ListNode {
    int val;
    ListNode next;
 
    ListNode() {
    }
 
    ListNode(int val) {
        this.val = val;
    }
 
    ListNode(int val, ListNode next) {
        this.val = val;
        this.next = next;
    }
}
 
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists.length == 0)
            return null;
 
        return merge(lists, 0, lists.length - 1);
    }
 
    public ListNode merge(ListNode[] lists, int low, int high) {
        if (high - low == 0)
            return lists[low];
        else if (high - low == 1)
            return mergeTwoLists(lists[low], lists[high]);
        else {
            int mid = (low + high) / 2;
            _____________________________;
            return mergeTwoLists(tmp1, tmp2);
        }
    }
 
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode head = new ListNode();
        ListNode p = head;
        while (l1 != null && l2 != null) {
            if (l1.val > l2.val) {
                p.next = l2;
                l2 = l2.next;
                p = p.next;
            } else {
                p.next = l1;
                l1 = l1.next;
                p = p.next;
            }
        }
        if (l1 != null)
            p.next = l1;
        if (l2 != null)
            p.next = l2;
        return head.next;
    }
}

解答思路：

以下是填补空白处的代码：

ListNode tmp1 = merge(lists, low, mid);
ListNode tmp2 = merge(lists, mid + 1, high);

完整的代码如下：

public class ListNode {
    int val;
    ListNode next;

    ListNode() {
    }

    ListNode(int val) {
        this.val = val;
    }

    ListNode(int val, ListNode next) {
        this.val = val;
        this.next = next;
    }
}

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists.length == 0)
            return null;

        return merge(lists, 0, lists.length - 1);
    }

    public ListNode merge(ListNode[] lists, int low, int high) {
        if (high - low == 0)
            return lists[low];
        else if (high - low == 1)
            return mergeTwoLists(lists[low], lists[high]);
        else {
            int mid = (low + high) / 2;
            ListNode tmp1 = merge(lists, low, mid);
            ListNode tmp2 = merge(lists, mid + 1, high);
            return mergeTwoLists(tmp1, tmp2);
        }
    }

    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode head = new ListNode();
        ListNode p = head;
        while (l1!= null && l2!= null) {
            if (l1.val > l2.val) {
                p.next = l2;
                l2 = l2.next;
                p = p.next;
            } else {
                p.next = l1;
                l1 = l1.next;
                p = p.next;
            }
        }
        if (l1!= null)
            p.next = l1;
        if (l2!= null)
            p.next = l2;
        return head.next;
    }
}

这段代码的思路是使用分治法来合并'K'个升序链表。'mergeKLists'方法首先判断链表数组是否为空，如果为空则返回'null'。然后调用'merge'方法进行合并。'merge'方法根据给定的范围将链表数组分成两部分，分别对这两部分进行合并，然后再将合并后的结果进行合并。'mergeTwoLists'方法用于合并两个升序链表。

(文章为作者在学习java过程中的一些个人体会总结和借鉴，如有不当、错误的地方，请各位大佬批评指正，定当努力改正，如有侵权请联系作者删帖。)