【LeetCode】合并K个升序链表

最新推荐文章于 2022-08-08 18:18:10 发布

xiaoshijiu333

最新推荐文章于 2022-08-08 18:18:10 发布

阅读量444

点赞数 1

分类专栏： LeetCode【链表】文章标签：链表 leetcode 数据结构

本文链接：https://blog.csdn.net/shijiujiu33/article/details/122642552

版权

LeetCode【链表】专栏收录该内容

15 篇文章 0 订阅

订阅专栏

#LeetCode每日一题【链表专题】

合并K个升序链表
https://leetcode-cn.com/problems/merge-k-sorted-lists/
分析
合并两个升序列表，之前已经实现过很多遍了；现在合并K个升序链表，给你一个链表数组，合并成有序的；
最容易想到的思路就是：一轮循环，记录结果两两合并
实现一

// 合并K个有序链表
    // 循环、拆分成合并2个有序列表
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) {
            return null;
        }
        ListNode dummy = null, node = null;
        for (ListNode list : lists) {
            node = mergeTwoList(dummy, list);
            dummy = node;
        }
        return node;
    }

    private ListNode mergeTwoList(ListNode list1, ListNode list2) {
        ListNode result = new ListNode(0);
        ListNode temp = result;
        ListNode node1 = list1, node2 = list2;
        while (node1 != null && node2 != null) {
            if (node1.val < node2.val) {
                temp.next = node1;
                node1 = node1.next;
            } else {
                temp.next = node2;
                node2 = node2.next;
            }
            temp = temp.next;
        }
        if (node1 != null) {
            temp.next = node1;
        }
        if (node2 != null) {
            temp.next = node2;
        }
        return result.next;
    }

LeetCode耗时：101ms 在这里插入图片描述

实现二
采用分治法：分而治之，不断地对半拆分，最后再合并；
归并排序也是这种思想

// 采用分治法：分而治之，不断地对半拆分，最后再合并
    // 归并排序也是这种思想
    public ListNode mergeKLists02(ListNode[] lists) {
        if (lists == null || lists.length == 0) {
            return null;
        }
        return mergeTwoList02(lists, 0, lists.length - 1);
    }

    // 递归：千万不要陷进去一轮轮跟着他走；
    // 递归三部曲：
    // 1. 结束条件是什么
    // 2. 返回值是什么
    // 3. 本级递归需要做什么

    /*
    例如每次递归：
    1、 结束条件：拆分到只有一个节点的时候，结束；在这边就是left=right，标记了数组的同一位置
    2、 返回值是什么：我们需要处理好的链表
    3、 本级递归应该做啥：本级递归，我们有左边拆分之后的链表、右边拆分之后的链表，那我们合并两个链表即可
     */
    private ListNode mergeTwoList02(ListNode[] lists, int left, int right) {
        if (left == right) {
            return lists[left];
        }
        int mid = (left + right) / 2;
        ListNode nodeLeft = mergeTwoList02(lists, left, mid);
        ListNode nodeRight = mergeTwoList02(lists, mid + 1, right);
        return mergeTwoList(nodeLeft, nodeRight);
    }
    private ListNode mergeTwoList(ListNode list1, ListNode list2) {
        ListNode result = new ListNode(0);
        ListNode temp = result;
        ListNode node1 = list1, node2 = list2;
        while (node1 != null && node2 != null) {
            if (node1.val < node2.val) {
                temp.next = node1;
                node1 = node1.next;
            } else {
                temp.next = node2;
                node2 = node2.next;
            }
            temp = temp.next;
        }
        if (node1 != null) {
            temp.next = node1;
        }
        if (node2 != null) {
            temp.next = node2;
        }
        return result.next;
    }