链表刷题之合并两个有序链表

1. 合并两个有序链表

  • 牛客
  • 力扣

将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

示例一:
输入:l1 = [1,2,4], l2 = [1,3,4]
输出:[1,1,2,3,4,4]

示例二:
输入:l1 = [], l2 = []
输出:[]

示例三:
输入:l1 = [], l2 = [0]
输出:[0]

题解:

哨兵节点的作用是方便得到最后的链表,即一个虚拟的头节点

链表刷题之合并两个有序链表_第1张图片

from typing import Optional


class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next


class Solution:
    def mergeTwoLists(self, list1: Optional[ListNode], list2: Optional[ListNode]) -> Optional[ListNode]:
        # 设置哨兵节点(虚拟头结点),将较小的节点连接到这个哨兵节点,最后返回 prehead.next 即可
        preHead = ListNode(-1)
        
        pre = preHead	# pre 指针,用于连接链表(指针或游标)
        l1 = list1
        l2 = list2

        while l1 and l2:
            # 将值较小的的节点接到 pre 指针
            if l1.val < l2.val:
                pre.next = l1
                l1 = l1.next
            else:
                pre.next = l2
                l2 = l2.next

            pre = pre.next  # 节点移动

        if l1:
            pre.next = l1

        if l2:
            pre.next = l2

        return preHead.next

参考:https://blog.csdn.net/weixin_43466639/article/details/124000412

你可能感兴趣的:(算法刷题,链表,数据结构)