合并两个排序的链表
描述
输入两个单调递增的链表,输出两个链表合成后的链表,当然我们需要合成后的链表满足单调不减规则。
算法
比较方便的做法是设置一个假的链表头 fake,而它的 next 指向的地址才是真正的表头,然后开始的时候将它作为 pre 指针。在循环中,每次都将要接入的新节点接到 pre 后面,然后更新 pre,保持这个循环不变式。
最后返回 fake 的 next 指向的地址,就是完整的结果。
// c++
class Solution {
public:
ListNode* Merge(ListNode* p1, ListNode* p2) {
ListNode* fake = new ListNode(0), *pre = fake;
while (p1 && p2) {
if (p1->val < p2->val) pre->next = p1, p1 = p1->next;
else pre->next = p2, p2 = p2->next;
pre = pre->next;
}
while (p1) pre->next = p1, p1 = p1->next, pre = pre->next;
while (p2) pre->next = p2, p2 = p2->next, pre = pre->next;
return fake->next;
}
};
class Solution:
def Merge(self, p1, p2):
fake = ListNode(0)
pre = fake
while p1 and p2:
if p1.val <= p2.val:
pre.next = p1
p1 = p1.next
else:
pre.next = p2
p2 = p2.next
pre = pre.next
while p1:
pre.next = p1
p1 = p1.next
pre = pre.next
while p2:
pre.next = p2
p2 = p2.next
pre = pre.next
return fake.next