LeetCode Remove Duplicates from Sorted List II

LeetCode解题之Remove Duplicates from Sorted List II

原题

把一个有序链表中所有重复的数字全部删光,删除后不再有原先重复的那些数字。

注意点:

  • 头节点也可能是重复的对象

例子:

输入: 1->1->2->3->3

输出: 2

解题思路

先按照 Remove Duplicates from Sorted List 的方法把重复的数字减至一个,接下来要考虑怎样可以把这一个节点也删除,那么只要在前一个节点对当前节点去重后,再把前一个节点的指针指向当前节点的后一个节点。因为要通过当前节点的前一个节点来操作,所以加了一个假的头节点。而如果节点没有重复的话不需要跳过当前节点,所以要一个额外的变量来记录当前节点是否有重复节点。

AC源码

# Definition for singly-linked list.
class ListNode(object):
    def __init__(self, x):
        self.val = x
        self.next = None

    def my_print(self):
        print(self.val)
        if self.next:
            print(self.next.val)


class Solution(object):
    def deleteDuplicates(self, head):
        """ :type head: ListNode :rtype: ListNode """
        dummy = ListNode(-1)
        dummy.next = head
        curr = dummy
        is_repeat = False
        while curr.next:
            while curr.next.next and curr.next.val == curr.next.next.val:
                curr.next = curr.next.next
                is_repeat = True
            if is_repeat:
                curr.next = curr.next.next
                is_repeat = False
            else:
                curr = curr.next
        return dummy.next


if __name__ == "__main__":
    n1 = ListNode(1)
    n2 = ListNode(1)
    n3 = ListNode(2)
    n1.next = n2
    n2.next = n3
    r = Solution().deleteDuplicates(n1)
    r.my_print()

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。

你可能感兴趣的:(LeetCode,算法,python,去重,链表)