剑指offer第18题:
题目一:在O(1)时间内删除链表节点(给点头节点与删除节点)
题目二:删除连续重复节点
class ListNode:
def __init__(self):
self.value = None
self.next_node = None
class Solution:
def list_generate(self, lst):
"""
生成链表
"""
if not lst:
return None
list_node = ListNode()
list_node.value = lst[0]
if len(lst) == 1:
list_node.next_node = None
else:
list_node.next_node = self.list_generate(lst[1:])
return list_node
def find_node(self, node, target_value):
"""
根据给定的目标值,找出指定节点的位置
非题目要求,只是为了测试验证
"""
if not target_value:
return False
while node:
if node.value == target_value:
return node
node = node.next_node
return False
def delete_node(self, head_node, del_node):
"""
删除指定节点
"""
if not (head_node and del_node):
return False
if del_node.next_node:
del_next_node = del_node.next_node
del_node.value = del_next_node.value
del_node.next_node = del_next_node.next_node
del_next_node.value = None
del_next_node.next_node =None
elif del_node == head_node:
head_node = None
del_node = None
else:
node = head_node
while node.next_node != del_node:
node = node.next_node
node.next_node = None
del_node = None
return head_node
def is_repeat(self, node):
"""
题目二:判断是否与后面的节点重复
"""
if node.next_node:
if node.value == node.next_node.value:
return True
return False
def delete_repeat_node(self, head_node):
"""
题目二:删除重复节点
"""
node = head_node
flag = False
"""
flag说明:例如'a'->'a',第一次循环删除第二个'a',然后通过这个flag来进行判断上一次操作是否删除重复节点
如果是的话,再次判断后面还有没有'a'了,没有,则通过这个flag把第一个'a'也应该删除,
"""
while node:
if solution.is_repeat(node):
head_node = self.delete_node(head_node, node)
flag = True
else:
if flag:
head_node = self.delete_node(head_node, node)
flag = False
node = node.next_node
return head_node
if __name__ == '__main__':
solution = Solution()
head_node = solution.list_generate(['a', 'a', 'a', 'b', 'c', 'e', 'd', 'd'])
head_node = solution.delete_repeat_node(head_node)
node = head_node
if node:
while node:
print node.value,
node = node.next_node
if node:
print '->',
else:
print 'wrong'