如何使用Python遍历链表(Linked List)

2023-04-11 00:00:00 遍历 链表 如何使用

首先,我们需要先定义链表节点的类:

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

其中,每个节点包含一个 val 属性表示节点存储的值,以及一个 next 属性表示下一个节点。

接下来,我们可以创建一个链表:

head = ListNode(1)
head.next = ListNode(2)
head.next.next = ListNode(3)

这个链表的结构为:

1 -> 2 -> 3 -> None

现在,我们可以通过遍历链表来访问所有节点。一个简单的方式是使用 while 循环:

node = head
while node:
    print(node.val)
    node = node.next

这里的 node 变量从链表的头节点 head 开始,每次更新为当前节点的下一个节点,直到达到链表的末尾节点。

完整代码:

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

head = ListNode(1)
head.next = ListNode(2)
head.next.next = ListNode(3)

node = head
while node:
    print(node.val)
    node = node.next

输出结果为:

1
2
3

相关文章