Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
从链表里删除指定内容的节点
class Node:
def __init__(self, v, n=None):
self.v = v
self.n = n
def do_print(head: Node):
t = head
print('=====')
while t is not None:
print(t.v)
t = t.n
print('=====')
def remove_nodes_with_val(head: Node, v):
tmp = Node(v+1, head)
p = tmp
while p.n is not None:
if p.n.v == v: # remove it
p.n = p.n.n
else:
p = p.n
return tmp.n
if __name__ == '__main__':
head = Node(1, Node(2, Node(3, Node(2, Node(1)))))
head = remove_nodes_with_val(head, 2)
do_print(head)
head = Node(1, Node(2, Node(3, Node(2, Node(1)))))
head = remove_nodes_with_val(head, 1)
do_print(head)
head = Node(1, Node(2, Node(3, Node(2, Node(1)))))
head = remove_nodes_with_val(head, 4)
do_print(head)
do_print(None)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment