「图解大厂面试高频算法题」链表专题-合并K个升序链表(归并思想)

本文链接：https://blog.csdn.net/Flames_and_Lights/article/details/121881210

「图解大厂面试高频算法题」链表专题-合并K个升序链表(归并思想)

PS: 本文为「图解大厂面试高频算法题」专题，主旨是根据“二八法则”的原理，以付出20%的时间成本，获得80%的刷题的收益，让那些想进互联网大厂的人少走些弯路。

PS: 欢迎关注我获取更多大厂面试总结。

原题链接: https://leetcode-cn.com/problems/merge-k-sorted-lists/

题目介绍

给你一个链表数组，每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例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

题目解答

这道题的解答依赖「图解大厂面试高频算法题」链表专题-合并两个有序链表，如果有不知道如何合并两个有序链表的同学们，请先复习一下哈。

两两合并(分而治之)

思路和算法

在这里插入图片描述
用分治的方法进行合并，将多个链表配对并将同一对中的链表进行合并，重复这一过程，直到我们得到了最终的有序链表。

初始状态
初始状态一共有六个链表。
在这里插入图片描述

第一次迭代
依次对链表对{j=0, j+i=1}，{j=2, j+i=3}，{j=4, j+i=5}中的链表进行合并，合并完之后我们还剩下6/2=3个链表需要继续合并。
在这里插入图片描述

第二次迭代
依次对链表对{j=0, j+i=2}，{j=4, j+i=6}中的链表进行合并，其中灰色节点并没有链表，也没有对链表对{j=4, j+i=6}进行合并，这里只是画了出来方便讲解。最终还剩下两条链表。
在这里插入图片描述

第三次迭代
最后对链表对{j=0, j+i=4}中的链表进行合并，最终只剩下一条链表，这条链表就是我们的答案。
在这里插入图片描述

代码实现

/**
 * Definition for singly-linked list.
 * 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 == null || lists.length == 0) {
            return null;
        }
        for (int i = 1; i < lists.length; i = i*2) {
            for (int j = 0; j + i < lists.length; j = j + i*2) {
                lists[j] = mergeTwoLists(lists[j], lists[j+i]);
            }
        }
        return lists[0];
    }
    ListNode mergeTwoLists(ListNode list1, ListNode list2) {
        ListNode dump = new ListNode();
        ListNode head = dump;
        while (list1 != null && list2 != null) {
            if (list1.val < list2.val) {
                head.next = list1;
                list1 = list1.next;
            } else {
                head.next = list2;
                list2 = list2.next;
            }
            head = head.next;
        }
        if (list1 != null) {
            head.next = list1;
        } else {
            head.next = list2;
        }
        return dump.next;
    }
}