题目:给定单链表的头指针和一个结点指针,定义一个函数在O(1)时间删除节点。
一般思路
从头开始顺序遍历单链表,遇到要删除的节点跳过去就行了
O(1)思路
因为已经给了要删除节点的指针,可以找到待删除节点的写一个节点的值,复制到待删除节点,将该节点的next指针指向next.next就行了,流程如下:
(1) a->b->c->d->e 要删除的是c节点
(2) a->b->d->d->e 将待删除节点的下一个节点的值复制到该节点
(3) a->b->d d->e 将待删除节点的next指向它的next.next
| ^
|_____|
要注意的问题
(1)要删除节点是头结点
(2)要删除节点是为节点(顺序遍历找到尾节点的上一个节点,并删除)
源码
#coding:utf-8
# 面试题13:在O(1)时间删除单链表节点
# 初始化单链表
class Node(object):
def __init__(self, val, p=0):
self.data = val
self.next = p
class LinkList(object):
def __init__(self):
self.head = 0
def initlist(self, data):
self.head = Node(data[0])
p = self.head
if data[0] == 'd':
self.target = self.head
for i in data[1:]:
node = Node(i)
p.next = node
p = p.next
if i == 'd':
self.target = p
return self.head, self.target # 生成完单链表返回头结点指针,目标删除指针
data = ['a','b','c','d','e']
l = LinkList()
listHead, deleteNode = l.initlist(data) # 头结点指针
def DeleteNode(listHead, deleteNode):
if listHead == deleteNode: # 要删除节点是头结点
listHead = listHead.next
elif deleteNode.next == None: # 要删除节点为尾节点
preNode = listHead
while True:
if preNode.next == deleteNode:
preNode.next = None
else:
preNode = preNode.next
else: # 要删除节点是中间节点
deleteNode.data = deleteNode.next.data
deleteNode.next = deleteNode.next.next