剑指Offer-Python-合并两个排序的链表

题目:合并两个排序的链表

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

思路:因为两个链表是有序的,将其合并,用到了双指针合并的方法。大致步骤:设置两个指针分别指向两个链表的头部,定义一个新的链表。比较两个指针结点大小,将较小结点插入新链表中,然后继续向后遍历两个链表,直到其中之一到尾部,新链表插入另一个非空链表。

代码:

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
class Solution:
    # 返回合并后列表
    def Merge(self, pHead1, pHead2):
        # write code here
        if pHead1 is None:
            return pHead2
        elif pHead2 is None:
            return pHead1
        p = pHead1
        q = pHead2
        h = new_head = ListNode(None)
        while p and q:
            if p.val <= q.val:
                h.next = p
                p = p.next
            else:
                h.next = q
                q = q.next
            h = h.next
        if p:
            h.next = p
        if q:
            h.next = q
        return new_head.next

猜你喜欢

转载自blog.csdn.net/qq_34364995/article/details/81288511