您现在的位置是:首页 >技术杂谈 >LeetCode:23. 合并 K 个升序链表网站首页技术杂谈

LeetCode:23. 合并 K 个升序链表

璃 白 2024-06-17 10:47:09
简介LeetCode:23. 合并 K 个升序链表

1)题目

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

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

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/merge-k-sorted-lists
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2)过程

先摸清 [[1,2,3],[4,5,6]] 的合并规律,start.val < lists[i].val
再摸清 [[4,5,6],[1,2,3]] 的合并规律,start.val > lists[i].val
最后摸清 [[-2],[-3,-2,1]] 的合并规律,start.val = lists[i].val

3)代码

1. 最开始

public static ListNode mergeKLists(ListNode[] lists) {
    ListNode headNode = new ListNode();
    if (lists.length == 0) return null;
    if (lists.length == 1) return lists[0];
    ListNode listNode;

    for (int i = 0; i < lists.length; i++) {
        listNode = headNode;
        if (listNode.next == null) {
            listNode.next = lists[i];
            continue;
        }

        while (lists[i] != null && listNode != null) {
            ListNode start = listNode.next;
            if (start == null) {
                listNode.next = lists[i];
                break;
            }

            if (start.val < lists[i].val) {
                // s后移一位
                listNode = start;
            } else {
                ListNode node = new ListNode(lists[i].val);
                //l移动到s前一位
                listNode.next = node;
                node.next = start;
                //h移动到node位置
                listNode = listNode.next;
                //l后移一位
                lists[i] = lists[i].next;
            }
        }
    }
    return headNode.next;
}

2.初步优化

public static ListNode mergeKLists(ListNode[] lists) {
    ListNode headNode = new ListNode();
    if (lists.length == 0) return null;
    if (lists.length == 1) return lists[0];
    ListNode listNode;

    for (int i = 0; i < lists.length; i++) {
        listNode = headNode;
        if (listNode.next == null) {
            listNode.next = lists[i];
            continue;
        }

        while (lists[i] != null && listNode != null) {
            ListNode start = listNode.next;
            if (start == null) {
                listNode.next = lists[i];
                break;
            }

            if (!(start.val < lists[i].val)) {
                ListNode node = new ListNode(lists[i].val);
                //l移动到s前一位
                listNode.next = node;
                node.next = start;
                //l后移一位
                lists[i] = lists[i].next;
            }

            listNode = listNode.next;
        }
    }
    return headNode.next;
}

4)结果

1. 最开始

在这里插入图片描述

2. 初步优化

在这里插入图片描述

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。