21. 合并两个有序链表
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4
解法一:非递归的做法,两个列表分别维护两个指针p1, p2。首先判断两个链表是否为空,如果一个链表为空则直接返回另一个链表即可。然后比较开头元素大小将比较小的用pHead指向作为最终返回值。接下来维护两个指针curP与preP,分别表示最终返回的链表的当前节点和当前节点的前一个节点。p1,p2不断比较大小,小的那一方如果之前的curP就是在本链表,则指针后移即可。如果之前curP不是在本链表,则需要curP和preP换另外的那个链表,另外的那个链表的指针后移。然后继续比较。最后记得将还有剩余值的链表接上。
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode *pHead = nullptr, *p1=l1, *p2=l2, *curP=nullptr, *preP=nullptr;
if (l1 == nullptr) return l2;
if (l2 == nullptr) return l1;
if (l1->val < l2->val) {
pHead = l1;
} else {
pHead = l2;
}
while (p1 != nullptr && p2 != nullptr) {
if (p1->val < p2->val) {
if (curP == nullptr || curP == p1) {
preP = p1;
p1 = p1->next;
curP = p1;
} else {
preP->next = p1;
preP = p1;
p1 = p1->next;
curP = p1;
}
} else {
if (curP == nullptr || curP == p2) {
preP = p2;
p2 = p2->next;
curP = p2;
} else {
preP->next = p2;
preP = p2;
p2 = p2->next;
curP = p2;
}
}
}
if (p1 == nullptr) {
preP->next = p2;
} else if (p2 == nullptr) {
preP->next = p1;
} else {
;
}
return pHead;
}
};
解法二:递归的做法,用解法一做完后看到题解,豁然开朗,直接用递归是如此简单。见代码:
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if(l1 == nullptr) {
return l2;
}
if(l2 == nullptr) {
return l1;
}
if(l1->val < l2->val) {
l1->next = mergeTwoLists(l1->next, l2);
return l1;
} else {
l2->next = mergeTwoLists(l1, l2->next);
return l2;
}
}
};