又是一道模拟题。这道题是要消除所有有重复值的节点,而且链表中的数是有序的。那么判断当前节点是否要删除,就是要判断当前节点是否和上一个节点的值相等,是否和下一个节点的值相等,满足一个相等条件,则删除这个节点。在本代码中,通过记录上一个节点的值来判断是否和上一个节点值相等(此处需要对表头特殊处理,因为表头没有上一指针。在本代码中,利用flag来标记是否是表头,若是表头,则不与上一节点值比较),通过p1.next is not None and p1.val == p1.next.val
来判断是否和下一个节点值相等。
整体代码如下:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def deleteDuplicates(self, head: ListNode) -> ListNode:
if head is None or head.next is None:
return head
he = ListNode(0)
he.next = head
p1 = he.next
p2 = he
last = head.val # 存储上一个node中的数
flag = 1 # 因为第一个节点没有前一个节点,所以将第一个节点用flag标记,为1则说明为第一个节点
while p1 is not None:
# 如果和前一个相等或者和后一个相等,则删除这个节点,同时更新last的值
if (p1.val == last and not flag) or (p1.next is not None and p1.val == p1.next.val):
last = p1.val
p1 = p1.next
p2.next = p1
# 如果没有重复的,那么将p1和p2都前进一格,同时更新last的值
else:
last = p1.val
p1 = p1.next
p2 = p2.next
if flag:
flag = 0
return he.next