LeetCode解题之Merge Two Sorted Lists
原题
将两个有序的链表拼接成一个有序的链表。
注意点:
- 不需要额外申请节点,主要把原链表中的节点串联起来
- 原链表中的一个已经全部在新链表中后,另一个链表剩余的部分可以直接拼接
例子:
输入: l1 = 1->2->4, l2 = 3
输出: 1->2->3->4
解题思路
为了避免分类讨论,添加一个假的头节点。现在只需要两个指针分别指向原来的两个链表,将其中比较小的节点添加到新的链表中。传入的参数l1和l2正好可以当作遍历两个链表的指针。
AC源码
# Definition for singly-linked list.
class ListNode(object):
def __init__(self, x):
self.val = x
self.next = None
class Solution(object):
def mergeTwoLists(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
temp = ListNode(-1)
head = temp
while l1 and l2:
if l1.val > l2.val:
temp.next = l2
l2 = l2.next
else:
temp.next = l1
l1 = l1.next
temp = temp.next
if l1:
temp.next = l1
else:
temp.next = l2
return head.next
if __name__ == "__main__":
assert Solution().mergeTwoLists(ListNode(1), ListNode(2)).val == 1
欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。