用Python写一个测链表长度的程序
链表是一种常见的数据结构,用于存储一系列元素。链表的长度是指链表中元素的数量,可以通过遍历链表并计数的方式来测量。下面是使用Python编写测链表长度程序的示例代码:
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
def get_length(head: ListNode) -> int:
"""
测量链表长度
:param head: 链表的头结点
:return: 链表的长度
"""
length = 0
while head:
length += 1
head = head.next
return length
在这段代码中,我们首先定义了一个ListNode类,用于表示链表的节点。每个节点包含一个val属性,表示节点的值,以及一个next属性,表示下一个节点的指针。
然后,我们定义了一个get_length函数,用于测量链表的长度。该函数接受一个链表的头结点作为参数,并返回链表的长度。在函数中,我们使用一个while循环遍历链表,每次遍历到一个节点就将长度加1。最后,我们返回计数器length的值,即链表的长度。
使用这个函数,我们可以很容易地测量任何链表的长度。例如,假设我们有以下链表:
1 -> 2 -> 3 -> 4 -> 5
我们可以按如下方式创建链表:
head = ListNode(1)
head.next = ListNode(2)
head.next.next = ListNode(3)
head.next.next.next = ListNode(4)
head.next.next.next.next = ListNode(5)
然后,我们可以调用get_length函数来测量链表的长度:
print(get_length(head)) # 输出:5
这将输出链表的长度,即5
原文地址: https://www.cveoy.top/t/topic/cHwn 著作权归作者所有。请勿转载和采集!