用python帮我实现一个链表,需要添加节点和查找下一个节点的功能
时间: 2024-03-29 18:40:05 浏览: 19
好的,我可以帮你实现一个简单的链表,以下是代码:
```python
class Node:
def __init__(self, data):
self.data = data
self.next = None
class LinkedList:
def __init__(self):
self.head = None
def add_node(self, data):
new_node = Node(data)
if not self.head:
self.head = new_node
else:
current = self.head
while current.next:
current = current.next
current.next = new_node
def find_next_node(self, data):
current = self.head
while current and current.data != data:
current = current.next
if current:
return current.next
else:
return None
```
以上代码定义了一个 `Node` 类表示链表中的节点,包含了一个数据 `data` 和一个指向下一个节点的指针 `next`。还定义了一个 `LinkedList` 类表示链表,包含了一个头节点 `head`,以及添加节点和查找下一个节点的方法。
你可以通过以下代码测试链表:
```python
my_list = LinkedList()
my_list.add_node(1)
my_list.add_node(2)
my_list.add_node(3)
print(my_list.find_next_node(2).data) # Output: 3
```
这里我们先创建一个空链表 `my_list`,然后向其中添加了三个节点,值分别为 1,2,3。最后查找了节点值为 2 的节点的下一个节点,输出了结果 3。