Leetcode刷题---21

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

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

输出：[1,1,2,3,4,4]

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
        if (list1 == nullptr) {
            return list2;
        } else if (list2 == nullptr) {
            return list1;
        } else if (list1 -> val < list2 -> val) {
            list1 -> next = mergeTwoLists(list1 -> next, list2);
            return list1;
        } else {
            list2 -> next = mergeTwoLists(list1, list2 -> next);
            return list2;
        }
    }
};

复杂度分析

时间复杂度：O(n+m)

空间复杂度：O(n+m)

方法二：迭代

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
        ListNode* preHead = new ListNode(-1);
        ListNode* prev = preHead;
        while (list1 != nullptr && list2 != nullptr) {
            if (list1 -> val < list2 -> val) {
                prev -> next = list1;
                list1 = list1 -> next;
            } else {
                prev -> next = list2;
                list2 = list2 -> next;
            }
            prev = prev -> next;
        }
        // 合并后list1和list2最多只有一个还没被合并完，直接将链表末尾指向未合并完的链表
        prev -> next = list1 == nullptr ? list2 : list1;
        return preHead -> next;
    }
};