题目描述:
给定一个单链表中的一个等待被删除的节点(非表头或表尾)。请在在O(1)时间复杂度删除该链表节点。
样例
给定 1->2->3->4,和节点 3,删除 3 之后,链表应该变为 1->2->4。
在单向链表中删除一个节点,常规的做法无疑是从链表的头节点开始,顺序遍历查找要删除的节点,并在链表中删除该节点。比如在如图3.3(a)所示的链表中,我们想删除节点i,可以从链表的头节点a开始顺序遍历,发现节点h的Next指向要删除的节点i,于是我们可以把节点h的Next指向i的下一个节点,即节点j。指针调整之后,我们就可以安全地删除节点i并保证链表没有断开,如图3.3(b)所示。这种思路由于需要顺序查找,时间复杂度自然就是O(n)了。
之所以需要从头开始查找,是因为我们需要得到将被删除的节点的前一个节点。在单向链表中,节点中没有指向前一个节点的指针,所以只好从链表的头节点开始顺序查找。
那是不是一定需要得到被删除的节点的前一个节点呢?答案是否定的。我们可以很方便地得到要删除的节点的下一个节点。如果我们把下一个节点的内容复制到需要删除的节点上覆盖原有的内容,再把下一个节点删除,那是不是就相当于把当前需要删除的节点删除了?
我们还是以前面的例子来分析这种思路。我们要删除节点i,先把i的下一个节点j的内容复制到i,然后把i的指针指向节点j的下一个节点。此时再删除节点j,其效果刚好是把节点i删除了,如图3.3(c)所示。
上述思路还有一个问题:如果要删除的节点位于链表的尾部,那么它就没有下一个节点,怎么办?我们仍然从链表的头节点开始,顺序遍历得到该节点的前序节点,并完成删除操作。
最后需要注意的是,如果链表中只有一个节点,而我们又要删除链表的头节点(也是尾节点),那么,此时我们在删除节点之后,还需要把链表的头节点设置为None。
"""
Definition of ListNode
class ListNode(object):
def __init__(self, val, next=None):
self.val = val
self.next = next
"""
class Solution:
# @param node: the node in the list should be deleted
# @return: nothing
def deleteNode(self, phead,pToBeDelete):
# write your code here
if phead == None or pToBeDelete == None:
return
# 删除非尾节点
if pToBeDelete.next != None:
tmpNode = pToBeDelete.next
pToBeDelete.val = tmpNode.val
pToBeDelete.next = tmpNode.next
# 只有一个节点
elif phead.val == pToBeDelete:
phead.val = None
phead.next = None
# 最后一种,删除节点是尾节点
else:
while phead.next != pToBeDelete:
phead = phead.next
phead.next = None