82 accepted, 15ms. The fastest is 6ms.

wufenggirl · Sep 6, 2017 · a31e550 · a31e550
1 parent c651843
commit a31e550
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/...ithms/0082.remove-duplicates-from-sorted-list-ii/remove-duplicates-from-sorted-list-ii.go b/...ithms/0082.remove-duplicates-from-sorted-list-ii/remove-duplicates-from-sorted-list-ii.go
@@ -7,6 +7,22 @@ type ListNode struct {
 }
 
 func deleteDuplicates(head *ListNode) *ListNode {
+	if head == nil {
+		return nil
+	}
+
+	if head.Next != nil && head.Val == head.Next.Val {
+		val := head.Val
+		head = head.Next.Next
+
+		for head != nil && head.Val == val {
+			head = head.Next
+		}
+
+		return deleteDuplicates(head)
+	}
+
+	head.Next = deleteDuplicates(head.Next)
 
 	return head
 }
diff --git a/.../0082.remove-duplicates-from-sorted-list-ii/remove-duplicates-from-sorted-list-ii_test.go b/.../0082.remove-duplicates-from-sorted-list-ii/remove-duplicates-from-sorted-list-ii_test.go
@@ -38,10 +38,10 @@ func Test_Problem0082(t *testing.T) {
 
 		question{
 			para{
-				[]int{1, 1, 2, 2, 3, 3, 4, 4, 5},
+				[]int{1, 1, 2, 2, 2, 2, 3, 3, 4, 4, 5, 6, 7, 8},
 			},
 			ans{
-				[]int{5},
+				[]int{5, 6, 7, 8},
 			},
 		},