合并两个有序的链表
输入两个单调递增的链表,输出两个链表合成后的链表,当然我们需要合成后的链表满足单调不减规则。
1. 递归版本
根据上面的图得到 , 先创建一个新的结点 , 然后比较两个链表的第一个节点 , 哪个节点的值小 , 就作为新的头节点
/*
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* ret = NULL;
if(pHead1->val <= pHead2->val)
{
ret = pHead1;
ret->next = Merge(pHead1->next, pHead2);
}
else
{
ret = pHead2;
ret->next = Merge(pHead1, pHead2->next);
}
return ret;
}
};
2. 非递归版本
两个指针, 分别指向两个链表的第一个节点, 比较两个节点的值的大小, 小的作为新的头结点, 然后指针后移, 再比较, 直到其中一个指向 NULL 为止, 然后再把另一个链表剩余的部分拼接到后面
/*
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* ret = NULL;
ListNode* tail = NULL;
if(pHead1->val <= pHead2->val)
{
ret = pHead1; // 先取较小节点作为新的头
pHead1 = pHead1->next;
}
else
{
ret = pHead2;
pHead2 = pHead2->next;
}
tail = ret; // tail 移动, ret 不动
while(pHead1 != NULL && pHead2 != NULL)
{
if(pHead1->val <= pHead2->val)
{
tail->next = pHead1; // 取较小节点加入合并链表中
pHead1 = pHead1->next;
tail = tail->next;
}
else
{
tail->next = pHead2;
pHead2 = pHead2->next;
tail = tail->next;
}
}
if(pHead1 != NULL)
{
tail->next = pHead1; // 如果有剩余, 接到后面
}
if(pHead2 != NULL)
{
tail->next = pHead2;
}
return ret;
}
};