[Leetcode 70] 82 Remove Duplicates from Sorted List II

Problems:

Given a sorted linked list, delete all nodes that have duplicate numbers, leaving only distinct numbers from the original list.

For example,
Given 1->2->3->3->4->4->5, return 1->2->5.
Given 1->1->1->2->3, return 2->3.

 

Analysis:

This problem is a bit different from the classic probelm of removing duplicates. Here we need to delete all nodes of duplication. To achieve this, three pointers and a flag are needed.

ListNode *ppre is used to point to the node that before dup;

ListNode *pre is used to point to the first node of the duplication;

ListNode *cur is used to point to the current checking node;

bool isDup is used to indicate whether there's duplication;

The checking process is as follows:

1. first check whther pre.val equals cur.vla or not; if equal go to step 2 else go to step 3

2. remove the cur node, and set the isDup flag; go back to step 1;

3. check whether isDup is set or not; if set, delete the pre node. otherwise move all three pointers one node further.

The space needed is constant and the time complexity is O(n).

 

One thing to remember is that after exitting the while loop, we need an extra check of isDup to decide whether delete or not the last node of the list

 

Code:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *deleteDuplicates(ListNode *head) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if (head == NULL || head->next == NULL)
            return head;
        
        ListNode dmy(-1);
        dmy.next = head;
        
        ListNode *ppre = &dmy, *pre = head, *cur = head->next;
        bool isDup = false;
        
        while (cur != NULL) {
            if (cur->val == pre->val) {
                isDup = true;
                pre->next = cur->next;
                cur->next = NULL;
                delete cur;
                cur = pre->next;
            } else {
                if (isDup) {
                    ppre->next = pre->next;
                    pre->next = NULL;
                    delete pre;
                    pre = cur;
                    cur = cur->next;
                    isDup = false;
                } else {
                    ppre = pre;
                    pre = cur;
                    cur = cur->next;
                }
            }
        }
        
        if (isDup) {
            ppre->next = pre->next;
            pre->next = NULL;
            delete pre;
        }
            
        return dmy.next;
    }
};