LeetCode – Remove Duplicates from Sorted List II (Java)

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.

 

解题思路：

对比I，这道题要求不保留任何重复的节点。所以需要一个index记录当前已经保留的节点的末端，然后用另一个节点去遍历，并在这个过程中对重复值和新值进行处理。

难点在于对corner cases的处理。刚开始刷题不久，还没有养成预先写test的习惯，所以在这些特例上，花了很多时间。

Corner Case：[1,2,2],[0,1,2,2,3,4],[1,1],[1],[1,1,2,2]

 

public static ListNode deleteDuplicates(ListNode head){
         if(head!=null){
             ListNode c1 = new ListNode(head.val-1);
             c1.next = head;
             ListNode re = c1, c2 = head;
             int count = 0, temp = c1.next.val;
             
             while(c2!=null){
                 if(temp!=c2.val && count>1){
                     c1.next = c2;  
                     temp = c2.val;
                     count = 1;
                 }else if(temp!=c2.val && count==1){
                     c1 = c1.next;
                     temp = c2.val;
                     count = 1;
                 }else 
                 count ++;
                 c2 = c2.next;
                  
         }
             if(count > 1) c1.next = null;
             return re.next;
         
     }
         return null;
}

 

 另外还有一个code更简洁的版本出自Programcreek，贴出来给大家参考。 http://www.programcreek.com/2014/06/leetcode-remove-duplicates-from-sorted-list-ii-java/

    public static ListNode deleteDuplicates(ListNode head){
            ListNode t = new ListNode(0);
            t.next = head;
            ListNode p = t;
            
            while(p.next!=null && p.next.next!=null){
                if(p.next.val == p.next.next.val){
                    int dup = p.next.val;
                    while(p.next!=null && p.next.val==dup){
                        p.next = p.next.next;
                    }
                }else
                    p=p.next;
            }          
            return t.next;
     }