从链表里删除指定内容的节点

remove_nodes_with_val_from_linkedlist.py

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)