链表查找值为val的元素python

原创

mob64ca12e04e7a 2023-08-27 11:59:19 ©著作权

文章标签 链表 python 编写代码 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e04e7a的原创作品，请联系作者获取转载授权，否则将追究法律责任

链表查找值为val的元素

简介

链表是一种重要的数据结构，它由一系列节点组成，每个节点包含一个数据元素和一个指向下一个节点的引用。在链表中查找特定值的节点是一种常见的操作。在本文中，我们将探讨如何使用Python编写代码来查找链表中特定值的节点。

链表实现

首先，我们需要实现一个链表类。链表的每个节点由一个值和一个指向下一个节点的引用（或指针）组成。下面是一个简单的链表节点类的实现：

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

查找链表中的特定值

要查找链表中特定值的节点，我们可以遍历整个链表，逐个检查每个节点的值，直到找到目标值或遍历到链表的末尾。下面是一个使用迭代方法实现的查找函数：

def search_val(head, val):
    curr = head
    while curr is not None:
        if curr.val == val:
            return curr
        curr = curr.next
    return None

在上述代码中，我们使用一个循环来遍历整个链表，每次迭代比较当前节点的值与目标值。如果找到匹配的节点，则返回该节点；如果遍历到链表的末尾仍未找到匹配的节点，则返回None。

测试案例

为了验证我们的查找函数的正确性，我们可以编写一些测试案例。下面是一个简单的测试案例：

# 创建一个链表: 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)

val = 3
result = search_val(head, val)
if result is not None:
    print(f"找到值为{val}的节点")
else:
    print(f"未找到值为{val}的节点")

以上测试案例创建了一个包含5个节点的链表，并通过调用search_val函数查找值为3的节点。如果找到了匹配的节点，则输出"找到值为3的节点"；否则输出"未找到值为3的节点"。