leetcode专题训练 82. Remove Duplicates from Sorted List II

又是一道模拟题。这道题是要消除所有有重复值的节点,而且链表中的数是有序的。那么判断当前节点是否要删除,就是要判断当前节点是否和上一个节点的值相等,是否和下一个节点的值相等,满足一个相等条件,则删除这个节点。在本代码中,通过记录上一个节点的值来判断是否和上一个节点值相等(此处需要对表头特殊处理,因为表头没有上一指针。在本代码中,利用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

你可能感兴趣的:(leetcode专题训练 82. Remove Duplicates from Sorted List II)