refactor 82

fishercoder1534 · fishercoder1534 · commit 9fbb85243867 · 2021-10-24T12:44:37.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_82.java b/src/main/java/com/fishercoder/solutions/_82.java
@@ -5,25 +5,21 @@
 public class _82 {
     public static class Solution1 {
         public ListNode deleteDuplicates(ListNode head) {
-            if (head == null) {
-                return head;
-            }
-            ListNode fakeHead = new ListNode(-1);
-            fakeHead.next = head;
-            ListNode pre = fakeHead;
-            ListNode curr = head;
-            while (curr != null) {
-                while (curr.next != null && curr.val == curr.next.val) {
-                    curr = curr.next;
+            ListNode pre = new ListNode(-1);
+            pre.next = head;
+            ListNode tmp = pre;
+            while (head != null) {
+                while (head.next != null && head.val == head.next.val) {
+                    head = head.next;
                 }
-                if (pre.next == curr) {
-                    pre = pre.next;
+                if (tmp.next == head) {
+                    tmp = tmp.next;
                 } else {
-                    pre.next = curr.next;
+                    tmp.next = head.next;
                 }
-                curr = curr.next;
+                head = head.next;
             }
-            return fakeHead.next;
+            return pre.next;
         }
     }
 
diff --git a/src/test/java/com/fishercoder/_82Test.java b/src/test/java/com/fishercoder/_82Test.java
@@ -9,26 +9,26 @@
 
 public class _82Test {
 
-  private static _82.Solution1 solution1;
-  private static ListNode head;
-  private static ListNode expected;
+    private static _82.Solution1 solution1;
+    private static ListNode head;
+    private static ListNode expected;
 
-  @BeforeClass
-  public static void setup() {
-    solution1 = new _82.Solution1();
-  }
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _82.Solution1();
+    }
 
-  @Test
-  public void test1() {
-    head = LinkedListUtils.contructLinkedList(new int[] {1, 2, 3, 3, 4, 4, 5});
-    expected = LinkedListUtils.contructLinkedList(new int[] {1, 2, 5});
-    Assert.assertEquals(expected, solution1.deleteDuplicates(head));
-  }
+    @Test
+    public void test1() {
+        head = LinkedListUtils.contructLinkedList(new int[]{1, 2, 3, 3, 4, 4, 5});
+        expected = LinkedListUtils.contructLinkedList(new int[]{1, 2, 5});
+        Assert.assertEquals(expected, solution1.deleteDuplicates(head));
+    }
 
-  @Test
-  public void test2() {
-    head = LinkedListUtils.contructLinkedList(new int[] {1, 1, 1, 2, 3});
-    expected = LinkedListUtils.contructLinkedList(new int[] {2, 3});
-    Assert.assertEquals(expected, solution1.deleteDuplicates(head));
-  }
+    @Test
+    public void test2() {
+        head = LinkedListUtils.contructLinkedList(new int[]{1, 1, 1, 2, 3});
+        expected = LinkedListUtils.contructLinkedList(new int[]{2, 3});
+        Assert.assertEquals(expected, solution1.deleteDuplicates(head));
+    }
 }
