移除双向链表中给定节点后的节点

class Node:
    def __init__(self, data):
        self.data = data
        self.prev = None
        self.next = None

def remove_node(head, node):
    # 判断给定节点是否为头节点
    if head == node:
        head = node.next
        if head:
            head.prev = None
    else:
        # 找到给定节点的前驱节点和后继节点
        prev = node.prev
        next = node.next
        prev.next = next
        if next:
            next.prev = prev

    return head