LeetCode 21. Merge Two Sorted Lists

162 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第9天,点击查看活动详情

LeetCode 21. Merge Two Sorted Lists

将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。 

 

示例 1:

输入: l1 = [1,2,4], l2 = [1,3,4]
输出: [1,1,2,3,4,4]

示例 2:

输入: l1 = [], l2 = []
输出: []

示例 3:

输入: l1 = [], l2 = [0]
输出: [0]

 

提示:

  • 两个链表的节点数目范围是 [0, 50]
  • -100 <= Node.val <= 100
  • l1 和 l2 均按 非递减顺序 排列

算法

(线性合并) O(n + m)
新建头部的保护结点 dummy,设置 cur 指针指向 dummy。
若当前 l1 指针指向的结点的值 val 比 l2 指针指向的结点的值 val 小,则令 cur 的 next 指针指向 l1,且 l1 后移;否则指向 l2,且 l2 后移。
然后 cur 指针按照上一部设置好的位置后移。
循环以上步骤直到 l1 或 l2 为空。
将剩余的 l1 或 l2 接到 cur 指针后边。\

1、l1表示当前链表1枚举到的位置,l2表示当前链表2枚举到的位置,dummy表示新开存结果的链表
2、当l1和l2都不为空
如果l1.val <= l2.val,则把l1.val加入到dummy链表的后面
如果l1.val > l2.val,则把l2.val加入到dummy链表的后面
3、当最后l1或者l2存在一个为空时,只有把存在剩余元素的链表继续拼接到dummy后面即可\

时间复杂度

两个链表各遍历一次,所以时间复杂度为 O(n + m)。

C++ 代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode *dummy = new ListNode(0);
        ListNode *cur = dummy;
        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;
        }

        cur -> next = (l1 != NULL ? l1 : l2);
        return dummy -> next;
    }
};