第三章 链表逆序输出
前言
本篇内容:简要介绍「链表逆序输出」的思路,并呈现例题、Python代码和注释。
链表逆序输出
「链表逆序输出」顾名思义,即为逆序地输出每个链表节点的值。不同于「链表反转」,需要保证链表的数据结构不变,逆序输出只需要保证最后的输出顺序即可。
LeetCode 图书整理 I
书店店员有一张链表形式的书单,每个节点代表一本书,节点中的值表示书的编号。为更方便整理书架,店员需要将书单倒过来排列,就可以从最后一本书开始整理,逐一将书放回到书架上。请倒序返回这个书单链表。
题目给我们一个链表,让我们倒序返回链表的节点值。
示例:
输入:head = [3,6,4,1]
输出:[1,4,6,3]
Definition 链表节点
首先定义链表节点,声明节点值val
和后继节点引用next
。
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
Solution1 顺序遍历,逆序输出
一种很自然的方法,遍历一遍已知的链表,将节点值按链表顺序依次存储到数组中,之后再将数组翻转即可。
Java和C++中的数组不是可变数组,如果使用数组存储,可以先遍历一遍链表获取长度,之后再定义数组。
考虑到可以用栈的"先入后出"来实现逆序,也可以选择LinkedList
或者stack
等变长容器来存储节点值。
Python中的数组[]
是可变长的,而切片操作可以直接返回翻转后的数组,实现逆序。
from typing import Optional, List
class Solution2: # Optional表示参数可以为ListNode也可以为None
def reverseBookList(self, head: Optional[ListNode]) -> List[int]:
stack = []
while head:
stack.append(head.val) # 顺序遍历存储节点值
head = head.next
return stack[::-1] # 切片操作 对所有元素进行倒序之后取出
Solution2 递归逆序拼接
递归也是逆序遍历的一种方法。
先递归到链表尾部,再依次存储节点值,从而实现链表的逆序输出。
class Solution1:
def reverseBookList(self, head: Optional[ListNode]) -> List[int]:
return self.reverseBookList(head.next) + [head.val] if head else []
self.reverseBookList(head.next)
会递归地遍历每一个节点。
return A if head else []
是一个三元表达式,如果head
存在则执行A,遍历到链表末尾时则会返回空列表,从而实现递归的终止条件。
[head.val]
是列表推导式,会创建一个包含当前head.val
值的列表。
+
是列表拼接运算符,最终按 [] + [n-1] + [n-2] + ... + [0]
的顺序拼接,实现逆序。
Test 测试
简单测试一下最终的结果。
head = ListNode(1)
head.next = ListNode(2)
head.next.next = ListNode(3)
head.next.next.next = ListNode(4) # 链表初始化 1->2->3->4->None
solution1 = Solution1()
solution2 = Solution2()
print(solution1.reverseBookList(head)) # 输出结果:[4, 3, 2, 1]
print(solution2.reverseBookList(head)) # 同上
总结
以上是今天的学习笔记,题记形式的文章会附上代码作为参考。
下期预告:删除链表节点
附:代码(可直接运行)
# 链表逆序输出
from typing import Optional, List
# Definition
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
# Solution1
class Solution1:
def reverseBookList(self, head: Optional[ListNode]) -> List[int]:
return self.reverseBookList(head.next) + [head.val] if head else []
# Solution2
class Solution2:
def reverseBookList(self, head: Optional[ListNode]) -> List[int]:
stack = []
while head:
stack.append(head.val)
head = head.next
return stack[::-1]
# Test
head = ListNode(1)
head.next = ListNode(2)
head.next.next = ListNode(3)
head.next.next.next = ListNode(4)
solution1 = Solution1()
solution2 = Solution2()
print(solution1.reverseBookList(head))
print(solution2.reverseBookList(head))