合并两个排序的链表

• 描述

输入两个递增的链表，单个链表的长度为n，合并这两个链表并使新链表中的节点仍然是递增排序的。

 

数据范围： 0≤n≤1000，−1000≤节点值≤1000
要求：空间复杂度 O(1)，时间复杂度 O(n)

 

如输入{1,3,5},{2,4,6}时，合并后的链表为{1,2,3,4,5,6}，所以对应的输出为{1,2,3,4,5,6}

 

• 示例1

输入：{1,3,5},{2,4,6}

返回值：{1,2,3,4,5,6}

 

• 示例2

输入：{},{}

返回值：{}

 

• 算法思想

1、先进行特殊情况的判断，若有一个链表为空，则返回另一个链表。

2、然后再判断哪个链表的头结点的值小，将头结点值大的链表合并至头结点值小的链表当中。

3、最后，若头结点值小的链表已到尾部，而头结点值大的链表仍有元素，就将剩下元素直接放到目标链表的尾部，并返回相应的目标链表。

 

• 代码

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
            val(x), next(NULL) {
    }
};*/
class Solution {
  public:
    ListNode* Merge(ListNode* pHead1, ListNode* pHead2) {
        ListNode* pre = nullptr;
        ListNode* p = pHead1;
        ListNode* q = pHead2;
        //特殊情况判断
        if(pHead1==nullptr)  return pHead2;
        if(pHead2==nullptr)  return pHead1;
        //L2合并至L1
        if (q->val >= p->val) {
            while (p && q) {
                if (q->val >= p->val) {
                    pre = p;
                    p = pre->next;
                } else {
                    ListNode* r = q->next;
                    q->next = p;
                    pre->next = q;
                    pre = q;
                    q = r;
                }
            }

            if (q) {
                pre->next = q;
            }
            return pHead1;
        } else {          //L1合并至L2
            while (p && q) {
                if (p->val >= q->val) {
                    pre = q;
                    q = pre->next;
                } else {
                    ListNode* r = p->next;
                    p->next = q;
                    pre->next = p;
                    pre = p;
                    p = r;
                }
            }
            if (p) {
                pre->next = p;
            }
            return pHead2;
        }
    }
};