LeetCode题目记录
第21题:合并两个有序链表
1.题目链接
[https://leetcode-cn.com/problems/merge-two-sorted-lists/]
2.题目内容
将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:l1: 1->2->4, l2: 1->3->4
输出:1->1->2->3->4->4
3.解题思路
本题需要对两个链表的值依次比大小,
(1)若l1.val < l2.val,就将l1.val赋给新的空链表(node)首位,
(2)然后将l1.next赋给l1,
(3)将node.next赋给node,依次进行
(4)最后若l1先遍历结束,即l1为空,就将l2直接全部赋在node的后面,node.next = l2
4.代码实现
# 自己解法
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
new_ListNode = ListNode(None)
node = new_ListNode
while l1 and l2: # 如果l1或l2任何一个遍历结束,则停止循环
if l1.val > l2.val:
l2,node.next = l2.next,l2
else:
l1,node.next = l1.next,l1
node = node.next # 别忘了这一步
if l1:
node.next = l1
else:
node.next = l2
return new_ListNode.next # 开始ListNode有head,即None,这个位置的输出要去掉head。
#(乙)
简化代码和步骤
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
node = new_ListNode = ListNode(None)
while l1 and l2:
if l1.val > l2.val :
node.next, l2 = l2, l2.next
else:
node.next, l1 = l1, l1.next
node = node.next
node.next = l1 or l2 # 如果l1是空的,就不会指向, 排除空的
return new_ListNode.next