LeetCode-021. Merge Two Sorted Lists

时间:2021-05-27 04:13:11

1. 题目

Merge Two Sorted Lists

不开辟额外空间,合并两条升序链表

2. 分析

采用归并排序的思想
Merge Sorted Array 的不同只是数据结构由数组换成了链表,代码反而更简洁。

1:迭代版,遍历两个链表,依次比较即可
由于两链表的长度可能不同,所以最终可能会有一个链表先遍历结束
这时注意不要遗漏剩余链表的元素

2:递归版,当其中一个链表为空,就返回另一个
比较两个节点值大小,如果l1的小,那么对于l1的下一节点和l2调用递归函数,将返回值赋值给l1->next,然后返回l1;
否则,对l2的下一节点和l1调用递归函数,将返回值赋值给l2->next,然后返回l2
注意mergeTwoLists(l1->next, l2);与mergeTwoLists(l1, l2->next);不要写反了

3. 代码

1)

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l1 == NULL)
            return l2;
        if(l2 == NULL)
            return l1;

        ListNode dummy(INT_MIN);
        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;
    }
};

2)

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l1 == NULL)
            return l2;
        if(l2 == NULL)
            return l1;

        if(l1->val <= l2->val)
        {
            l1->next = mergeTwoLists(l1->next, l2);
            return l1;
        }
        else
        {
            l2->next = mergeTwoLists(l2->next, l1);
            return l2;
        }
    }
};

完整源代码放于github