Mean Mode

ardallie · ardallie · commit 35ae53bcfb94 · 2021-08-13T22:58:09.000+01:00
diff --git a/src/easy/MeanMode.java b/src/easy/MeanMode.java
@@ -0,0 +1,65 @@
+package easy;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Have the function MeanMode(arr) take the array of numbers stored in arr
+ * and return 1 if the mode equals the mean, 0 if they don't equal each other
+ * (i.e. [5, 3, 3, 3, 1] should return 1 because the mode (3) equals the mean (3)).
+ * The array will not be empty, will only contain positive integers,
+ * and will not contain more than one mode.
+ */
+public class MeanMode {
+
+  /**
+   * Mean Mode function.
+   *
+   * @param arr input array.
+   * @return 1 if the mode equals the mean, 0 if they don't equal each other
+   */
+  private static String meanMode(int[] arr) {
+    int sum = 0;
+    int modeKey = 0;
+    int modeVal = 0;
+
+    Map<Integer, Integer> modeMap = new HashMap<>();
+    for (int item : arr) {
+      modeMap.put(item, 0);
+    }
+
+    for (int value : arr) {
+      sum += value;
+      int val = modeMap.get(value);
+      if (val > 0) {
+        modeMap.put(value, val + 1);
+      } else {
+        modeMap.put(value, 1);
+      }
+    }
+
+    for (Map.Entry<Integer, Integer> item : modeMap.entrySet()) {
+      int itemKey = item.getKey();
+      int itemVal = item.getValue();
+      if (itemVal > modeVal) {
+        modeVal = itemVal;
+        modeKey = itemKey;
+      }
+    }
+
+    return modeKey == (sum / arr.length) ? "1" : "0";
+  }
+
+  /**
+   * Entry point.
+   *
+   * @param args command line arguments
+   */
+  public static void main(String[] args) {
+    var result1 = meanMode(new int[]{5, 3, 3, 3, 1});
+    System.out.println(result1);
+    var result2 = meanMode(new int[]{64, 64, 64, 64, 64, 64, 64, 64, 1024});
+    System.out.println(result2);
+  }
+
+}
