合并K个排序链表

写在前面

写这个是之前听说有人猿辅导面试让手撕这道题,不能用循环,这就逼着大家用递归来做。如果换做是我的话肯定也十分困难,因为我之前做的是用迭代做的,没有仔细看递归,劝大家有多法的题一定要多看看,尽量掌握。(庆幸没遇到这个题)今日记录一下。

合并两个有序链表

先别一口吃个大胖子,先从简单的开始,合并两个排序好的链表。来自LeetCode21
题目描述:将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

示例:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4

题解及思路

迭代应该很好想,先新建一个哨兵,依次判断两个节点的值的大小,小的连接到哨兵后边,指针后移,不断循环。

public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode pre = new ListNode(-1);
        ListNode cur = pre;
        while (l1 != null && l2 != null) {
            if (l1.val < l2.val) {
                cur.next = l1;
                l1 = l1.next;
            } else {
                cur.next = l2;
                l2 = l2.next;
            }
            cur = cur.next;
        }
        if (l1 == null) {
            cur.next = l2;
        } else {
            cur.next = l1;
        }
        return pre.next;

递归写法,一般写递归都要先考虑递归终止条件,容易发现,当一个节点为null时那么就可以直接返回另一个节点即可。后边的逻辑就是判断两个节点的值,修改小的节点指针指向,递归返回。代码如下:

 /**
     * 递归合并两个排序链表
     *
     * @param list1
     * @param list2
     * @return
     */
    private ListNode merge2ListNode(ListNode list1, ListNode list2) {
        if (list1 == null) return list2;
        if (list2 == null) return list1;
        if (list1.val < list2.val) {
            list1.next = merge2ListNode(list1.next, list2);
            return list1;
        } else {
            list2.next = merge2ListNode(list1, list2.next);
            return list2;
        }
    }

合并K个有序链表

有了合并两个排序链表的思路,那么合并k个的话自然可以想到迭代两个合成一个,然后再往后边合并。这里也可用递归的办法进行合并。类似归并排序的思想,不断切分,然后借助于合并两个链表的函数进行归并。

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 low, int high) {
        if (low == high) return lists[low];
        int mid = (low + high) >> 1;
        ListNode l1 = merge(lists, low, mid);
        ListNode l2 = merge(lists, mid + 1, high);
        return merge2ListNode(l1, l2);
    }

最后还是想说一句,方法能尽量多掌握就多掌握。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值