在O(1)的时间内删除链表节点

题目:


在O(1)的时间内删除链表节点。给定链表的头指针和待删除的节点指针,定义一个函数在O(1)的时间内删除该节点。

剑指offer的思路,顿时觉得极妙。删除节点node1,先把其下一个节点node2的内容复制到node1这个节点,然后把节点node1的指针指向node2的下一个节点,最后删除节点node2。

解题:

def delete(link, node):
    if node.next:  # 删除节点不是最后一个节点,此时链表也必不可能只有一个节点
        p = node.next
        node.val = p.val
        node.next = p.next
        del p
    elif link == node:  # 链表只有一个节点。(删除的节点即是头2节点也是尾节点)
        del node
        link = None
    else:   # 其他情况,即链表长度大于1且删除的是尾节点
        p = link
        while p.next != node:
            p = p.next
        p.next = None
        del node
posted @ 2019-05-28 20:37  长安223  阅读(139)  评论(0编辑  收藏  举报