合并两个排序的链表

描述

输入两个单调递增的链表,输出两个链表合成后的链表,当然我们需要合成后的链表满足单调不减规则。

算法

比较方便的做法是设置一个假的链表头 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
posted @ 2021-06-23 16:39  大菠萝侦探  阅读(25)  评论(0编辑  收藏  举报