题目描述:
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 1:
输入:l1 = [1,2,4], l2 = [1,3,4] 输出:[1,1,2,3,4,4]
示例 2:
输入:l1 = [], l2 = [] 输出:[]
示例 3:
输入:l1 = [], l2 = [0] 输出:[0]s
解题思路:这个题的逻辑比较简单,两个链表首先是有序的,所以每次选择节点插入新的链表时,只需要比较一次就可以。循环的判定是当两个链表一个为空就结束(因为有序,所以剩下的一个链表只需要直接接在新的链表后面就行),每次循环做一次比较,比较后将指针向后挪一位。
有一个注意的地方是首先定义一个虚拟节点,也就是 dummy
节点。如果不使用 dummy
虚拟节点,代码会复杂一些,需要额外处理指针 p
为空的情况。而有了 dummy
节点这个占位符,可以避免处理空指针的情况,降低代码的复杂性。
还有一个一开始写错的地方,最后返回一开始写成了return p,报错后发现p只是一个指针,而链表则是dummy.next
还有就是链表无需进行初始化,只需要定义一个指针找到第一个节点的位置即可
代码:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def mergeTwoLists(self, list1, list2):
"""
:type list1: Optional[ListNode]
:type list2: Optional[ListNode]
:rtype: Optional[ListNode]
"""
dummy = ListNode(-1)
p = dummy
p1 = list1
p2 = list2
while (p1 and p2):
if (p1.val < p2.val):
p.next =p1
p1 = p1.next
else:
p.next = p2
p2 = p2.next
p = p.next
if (p1 ):
p.next = p1
if (p2):
p.next = p2
return dummy.next