算法学习-BM4:合并两个排序的链表

算法学习

BM4 合并两个排序的链表

https://www.nowcoder.com/practice/a479a3f0c4554867b35356e0d57cf03d?tpId=295&sfm=html&channel=nowcoder

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

思路：

归并排序：每次比较两个头部，从中取出最小的元素，然后依次往后。
1.判断空链表的情况，只要有一个链表为空，那答案必定就是另一个链表了，就算另一个链表也为空。
2.新建一个空的表头后面连接两个链表排序后的结点。
3.遍历两个链表都不为空的情况，取较小值添加在新的链表后面，每次只把被添加的链表的指针后移。
4.遍历到最后肯定有一个链表还有剩余的结点，它们的值将大于前面所有的，直接连在新的链表后面即可。

代码实现：

#include <iostream>

//定义链表
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};


class Solution {
public:
	ListNode* Merge(ListNode* pHead1, ListNode* pHead2) {
		if (pHead1==NULL) //有一个为空，返回另外一个
		{
			return pHead2;
		}
		if (pHead2 == NULL) 
		{
			return pHead1;
		}
		ListNode* head = new ListNode(0); //新表头，以便输出
		ListNode* cur = head;  //新链表的指针
		while (pHead1 && pHead2) //两个链表都不为空
		{
			if (pHead1->val <= pHead2->val) //取小val的节点
			{
				cur->next = pHead1; // 小的val节点 添加在新表后
				pHead1 = pHead1->next;  //移动取值指针
			}
			else
			{
				cur->next = pHead2; // 小的val节点 添加在新表后
				pHead2 = pHead2->next;  //移动取值指针
			}
			cur = cur->next; //新表指针后移
		}
		if (pHead1)  //哪个链表还有剩余，直接连接在新表后
		{
			cur->next = pHead1;
		}
		else
		{
			cur->next = pHead2;
		}
		return head->next; //返回值去掉表头
	}
};