《图解算法数据结构》学习笔记3——链表逆序输出

---

前言

本篇内容：简要介绍「链表逆序输出」的思路，并呈现例题、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))