leetcode hot 100

合并两个有序列表

leetcode hot 100_第1张图片
在本题中,我们要把两个升序的列表合并成一个新的升序的列表,同时要求空间复杂度为O(1),那么这时候我们就不能新建一个列表来储存结果,需要在原始列表上进行操作。

那我们直接可以设置一个虚拟节点,让它保持在最开始节点之前,以便排序完找到头节点。同时再设置一个指针,此时指针指向虚拟节点。之后我们开始比较两个链表的值,如果当年链表1的值要比链表2的值小,那么我们排序的话应该先排链表1,那么我们直接让指针指向链表1当前节点,然后把链表1往后走一步。之后再把指针往后走一步。

以此类推,继续移动,当移动结束后,可能两个链表其中一个还没有结束,那由于它本身是升序的,直接把指针指向当前链表的节点即可。

class Solution {
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        // 一开始设置一个虚拟节点,它的值为 -1,它的值可以设置为任何的数,因为我们根本不需要使用它的值
        ListNode dummy = new ListNode(-1);

        // 设置一个指针,指向虚拟节点
        ListNode pre = dummy;

        // 通过一个循环,不断的比较 l1 和 l2 中当前节点值的大小,直到 l1 或者 l2 遍历完毕为止
        while (l1 != null && l2 != null) {
            // 如果 l1 当前节点的值小于等于了 l2 当前节点的值
            if (l1.val <= l2.val) {
                // 让 pre 指向节点的 next 指针指向这个更小值的节点
                // 即指向 l1
                pre.next = l1;
                // 让 l1 向后移动
                l1 = l1.next;
            }else {
                // 让 pre 指向节点的 next 指针指向这个更小值的节点
                // 即指向 l2
                pre.next =l2;
                // 让 l2 向后移动
                l2 = l2.next;
            }
            // 让 pre 向后移动
            pre = pre.next;
        }

        // 跳出循环后,l1 或者 l2 中可能有剩余的节点没有被观察过
        // 直接把剩下的节点加入到 pre 的 next 指针位置

        // 如果 l1 中还有节点
        if (l1 != null) {
            // 把 l1 中剩下的节点全部加入到 pre 的 next 指针位置
            pre.next = l1;
        }

        // 如果 l2 中还有节点
        if (l2 != null) {
            // 把 l2 中剩下的节点全部加入到 pre 的 next 指针位置
            pre.next = l2;
        }

        // 最后返回虚拟节点的 next 指针
        return dummy.next;
    }
}

来源:思路解法代码

你可能感兴趣的:(leetcode,算法,职场和发展)