refactor 75

fishercoder1534 · fishercoder1534 · commit f986b8aeb870 · 2021-10-12T11:25:45.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_75.java b/src/main/java/com/fishercoder/solutions/_75.java
@@ -4,13 +4,13 @@ public class _75 {
 
     public static class Solution1 {
         public void sortColors(int[] nums) {
-            int zero = 0;
-            int two = nums.length - 1;
-            for (int i = 0; i <= two; ) {
-                if (nums[i] == 0 && i > zero) {
-                    swap(nums, i, zero++);
-                } else if (nums[i] == 2 && i < two) {
-                    swap(nums, i, two--);
+            int zeroIndex = 0;
+            int twoIndex = nums.length - 1;
+            for (int i = 0; i <= twoIndex; ) {
+                if (nums[i] == 0 && i > zeroIndex) {
+                    swap(nums, i, zeroIndex++);
+                } else if (nums[i] == 2 && i < twoIndex) {
+                    swap(nums, i, twoIndex--);
                 } else {
                     i++;
                 }
diff --git a/src/test/java/com/fishercoder/_75Test.java b/src/test/java/com/fishercoder/_75Test.java
@@ -7,46 +7,104 @@
 import static org.junit.Assert.assertArrayEquals;
 
 public class _75Test {
-  private static _75.Solution1 solution1;
-  private static int[] nums;
-
-  @BeforeClass
-  public static void setup() {
-    solution1 = new _75.Solution1();
-  }
-
-  @Test
-  public void test1() {
-    nums = new int[] {2, 2, 1};
-    solution1.sortColors(nums);
-    assertArrayEquals(new int[] {1, 2, 2}, nums);
-  }
-
-  @Test
-  public void test2() {
-    nums = new int[] {0, 1, 2, 0, 2, 1};
-    solution1.sortColors(nums);
-    assertArrayEquals(new int[] {0, 0, 1, 1, 2, 2}, nums);
-  }
-
-  @Test
-  public void test3() {
-    nums = new int[] {0};
-    solution1.sortColors(nums);
-    assertArrayEquals(new int[] {0}, nums);
-  }
-
-  @Test
-  public void test4() {
-    nums = new int[] {1, 0};
-    solution1.sortColors(nums);
-    assertArrayEquals(new int[] {0, 1}, nums);
-  }
-
-  @Test
-  public void test5() {
-    nums = new int[] {2};
-    solution1.sortColors(nums);
-    assertArrayEquals(new int[] {2}, nums);
-  }
+    private static _75.Solution1 solution1;
+    private static int[] nums;
+
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _75.Solution1();
+    }
+
+    @Test
+    public void test1() {
+        nums = new int[]{2, 2, 1};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{1, 2, 2}, nums);
+    }
+
+    @Test
+    public void test2() {
+        nums = new int[]{0, 1, 2, 0, 2, 1};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{0, 0, 1, 1, 2, 2}, nums);
+    }
+
+    @Test
+    public void test3() {
+        nums = new int[]{1, 0, 2};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{0, 1, 2}, nums);
+    }
+
+    @Test
+    public void test4() {
+        nums = new int[]{1, 0};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{0, 1}, nums);
+    }
+
+    @Test
+    public void test5() {
+        nums = new int[]{2};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{2}, nums);
+    }
+
+    @Test
+    public void test6() {
+        nums = new int[]{2, 0, 1};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{0, 1, 2}, nums);
+    }
+
+    @Test
+    public void test7() {
+        nums = new int[]{0};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{0}, nums);
+    }
+
+    @Test
+    public void test8() {
+        nums = new int[]{2, 2};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{2, 2}, nums);
+    }
+
+    @Test
+    public void test9() {
+        nums = new int[]{2};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{2}, nums);
+    }
+
+    @Test
+    public void test10() {
+        nums = new int[]{1};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{1}, nums);
+    }
+
+    @Test
+    public void test11() {
+        nums = new int[]{1, 2, 0};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{0, 1, 2}, nums);
+    }
+
+    @Test
+    public void test12() {
+        nums = new int[]{1, 2, 1};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{1, 1, 2}, nums);
+    }
+
+    @Test
+    public void test13() {
+        nums = new int[]{2, 0, 2, 1, 1, 0};
+        solution1.sortColors(nums);
+        assertArrayEquals(new int[]{0, 0, 1, 1, 2, 2}, nums);
+    }
+
+
 }
