refactor 575

fishercoder1534 · fishercoder1534 · commit d119cdf6785e · 2021-03-01T07:34:11.000-08:00
diff --git a/src/main/java/com/fishercoder/solutions/_575.java b/src/main/java/com/fishercoder/solutions/_575.java
@@ -1,24 +1,16 @@
 package com.fishercoder.solutions;
 
-import java.util.Arrays;
+import java.util.HashSet;
+import java.util.Set;
 
 public class _575 {
     public static class Solution1 {
-        public int distributeCandies(int[] candies) {
-            Arrays.sort(candies);
-            int sisCount = 0;
-            for (int i = 0; i < candies.length; i++) {
-                int val = candies[i];
-                sisCount++;
-                if (sisCount >= candies.length / 2) {
-                    return candies.length / 2;
-                }
-                while (i < candies.length && candies[i] == val) {
-                    i++;
-                }
-                i--;
+        public int distributeCandies(int[] candyType) {
+            Set<Integer> map = new HashSet<>();
+            for (int candy : candyType) {
+                map.add(candy);
             }
-            return sisCount;
+            return Math.min(map.size(), candyType.length / 2);
         }
     }
 }
diff --git a/src/test/java/com/fishercoder/_575Test.java b/src/test/java/com/fishercoder/_575Test.java
@@ -6,14 +6,11 @@
 
 import static org.junit.Assert.assertEquals;
 
-/**
- * Created by fishercoder on 5/6/17.
- */
 public class _575Test {
     private static _575.Solution1 solution1;
     private static int expected;
     private static int actual;
-    private static int[] candies;
+    private static int[] candyType;
 
     @BeforeClass
     public static void setup() {
@@ -22,25 +19,25 @@ public static void setup() {
 
     @Test
     public void test1() {
-        candies = new int[]{1, 1, 2, 3};
+        candyType = new int[]{1, 1, 2, 3};
         expected = 2;
-        actual = solution1.distributeCandies(candies);
+        actual = solution1.distributeCandies(candyType);
         assertEquals(expected, actual);
     }
 
     @Test
     public void test2() {
-        candies = new int[]{1, 1, 2, 2, 3, 3};
+        candyType = new int[]{1, 1, 2, 2, 3, 3};
         expected = 3;
-        actual = solution1.distributeCandies(candies);
+        actual = solution1.distributeCandies(candyType);
         assertEquals(expected, actual);
     }
 
     @Test
     public void test3() {
-        candies = new int[]{1000, 1, 1, 1};
+        candyType = new int[]{1000, 1, 1, 1};
         expected = 2;
-        actual = solution1.distributeCandies(candies);
+        actual = solution1.distributeCandies(candyType);
         assertEquals(expected, actual);
     }
 }
