hashmap good

src/kdiv/Distribution.java

+package kdiv;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * The Distribution class manages a collections of Objects and their associated values.
+ */
+public class Distribution {
+    private Map<Object, Integer> objectMap;
+
+    /**
+     *  Initialization
+     */
+    public Distribution() {
+        this.objectMap = new HashMap<>();
+    }
+
+    /**
+     * Adds an object with an associated integer value.
+     * @param obj The object to store.
+     * @param value The integer value associated with the object.
+     */
+    public void addObject(Object obj, int value) {
+        objectMap.put(obj, value);
+    }
+
+    /**
+     * Retrieves the value of a given object.
+     * @param obj The object to look up.
+     * @return The associated integer value, or null if not found.
+     */
+    public Integer getValue(Object obj) {
+        return objectMap.getOrDefault(obj, null);
+    }
+
+    /**
+     * Checks if the object exists in the distribution.
+     * @param obj The object to check.
+     * @return True if the object exists, false otherwise.
+     */
+    public boolean contains(Object obj) {
+        return objectMap.containsKey(obj);
+    }
+
+    /**
+     * Gets the entire distribution as a HashMap.
+     * @return The HashMap storing object-value pairs.
+     */
+    public Map<Object, Integer> getObjectMap() {
+        return objectMap;
+    }
+
+    @Override
+    public String toString() {
+        return objectMap.toString();
+    }
+}