在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