合并两个排序的链表C++

题目描述

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

如输入{1,3,5},{2,4,6}时，合并后的链表为{1,2,3,4,5,6}，所以对应的输出为{1,2,3,4,5,6}，转换过程如下图所示：

或输入{-1,2,4},{1,3,4}时，合并后的链表为{-1,1,2,3,4,4}，所以对应的输出为{-1,1,2,3,4,4}，转换过程如下图所示：

示例1：

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

示例2：

输  入：{},{}
返回值：{}

示例3：

输  入：{-1,2,4},{1,3,4}
返回值：{-1,1,2,3,4,4}

代码

class Solution {
  public:
    ListNode* Merge(ListNode* pHead1, ListNode* pHead2) {//定义两个头节点指针
        if (pHead1 == NULL) {
            return pHead2;
        } else if (pHead2 == NULL) {
            return pHead1;
        }

        ListNode* MergeHead = NULL;//初始化合并头节点
        if (pHead1->val < pHead2->val) {//pHead1->val = 1 < pHead2->val=2,合并头节点的值为1，继续寻找下一个节点。
            MergeHead = pHead1;
            pHead1 = pHead1->next;
        } else {
            MergeHead = pHead2;
            pHead2 = pHead2->next;
        }
        ListNode* tmpNode = MergeHead;
        while (pHead1 && pHead2) {
            if (pHead1->val < pHead2->val) {//当两个链表的值都存在时，pHead1的值小于pHead2的值，就将值小的赋给合并节点的下一个节点，继续遍历链表
                MergeHead->next = pHead1;
                pHead1 = pHead1->next;
            } else {
                MergeHead->next = pHead2;
                pHead2 = pHead2->next;
            }
            MergeHead = MergeHead->next;
        }
        //如果链表中的值仅有一个，则直接赋值，为空也直接赋值
        if (pHead1) {
            MergeHead->next = pHead1;
        }
        if (pHead2) {
            MergeHead->next = pHead2;
        }
		return tmpNode;
    };
};