21. 合并两个有序链表

leetcode21:21. 合并两个有序链表

题目描述

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

Example

输入：1->2->4, 1->3->4
输出：1->1->2->3->4->4

solution idea

/**
 * 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) {
        // 处理空链表
        if (l1==NULL) return l2;
        if (l2==NULL) return l1;

        // 保证l1中的首结点为最终链表的头节点
        if(l1 -> val > l2 -> val) swap(l1, l2);

        ListNode* res=l1;
        ListNode* temp=nullptr;
        while(l1->next!=NULL && l2!=NULL)
        {
            // l1中节点小于l2
            while(l1->next!=NULL && l1->next->val <= l2->val)
            {
                l1=l1->next;
            }
            if(l1 -> next == NULL) break;
            // l2 中节点小于l1 ,将l2 node 插入到 l1 对应序列
            while(l2!=NULL && l1->next->val > l2->val)
            {
                temp=l2;
                l2=l2->next;
                temp->next=l1->next;
                l1->next=temp;
            }
        }
        // 若 l1 非空，则自然接在链表尾端，若 l2 非空，则将当前头节点接在 l1 尾端
        if (l2!=NULL)   l1->next=l2;
        return res;
    }
};

参考文献

1. c++ prime 第5版