Leetcode 21. 合并两个有序链表

题目描述

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

 

示例 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 均按 非递减顺序 排列

---

解答

不想寫recursive programming，傳統的while loop消耗更低。 1. 兩個list都有的時候 2. 一個空了一個沒空 3. 兩格都空了才算結束

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
        ListNode* ans=nullptr,*now=ans,*ptr1=list1,*ptr2=list2;
        while(ptr1!=nullptr && ptr2!=nullptr){
            int to_append;
            if (ptr1->val<=ptr2->val){
                to_append=ptr1->val;
                ptr1=ptr1->next;
            }
            else{
                to_append=ptr2->val;
                ptr2=ptr2->next;
            }
            if (ans==nullptr){
                ans=new ListNode(to_append);
                now=ans;
            }
            else{
                now->next=new ListNode(to_append);
                now=now->next;
            }
        }
        if (ptr1==nullptr&& ptr2==nullptr){
            return ans;
        }
        else{
            if (ptr1==nullptr){
                ptr1=ptr2;
            }
            while(ptr1!=nullptr){
                if (ans==nullptr){
                    ans=new ListNode(ptr1->val);
                    now=ans;
                }
                else{
                    now->next=new ListNode(ptr1->val);
                    now=now->next;
                }                
                
                ptr1=ptr1->next;
                
            }
        }
        return ans;
    }
};

Complexity

• Time: \(O(N)\)
• Space: \(O(1)\)
  • 不考慮輸入輸出的空間