Create Design a Number Container System.java

varunu28 · web-flow · commit aa8154baf0af · 2025-02-08T11:35:50.000-08:00
diff --git a/Medium/Design a Number Container System.java b/Medium/Design a Number Container System.java
@@ -0,0 +1,34 @@
+class NumberContainers {
+
+    private final Map<Integer, TreeSet<Integer>> numberToIndex;
+    private final Map<Integer, Integer> indexToNumber;
+
+    public NumberContainers() {
+        this.numberToIndex = new HashMap<>();  
+        this.indexToNumber = new HashMap<>();      
+    }
+    
+    public void change(int index, int number) {
+        if (indexToNumber.containsKey(index)) {
+            int prevNumber = indexToNumber.get(index);
+            numberToIndex.get(prevNumber).remove(index);
+        }
+        indexToNumber.put(index, number);
+        numberToIndex.computeIfAbsent(number, k -> new TreeSet<>()).add(index);
+    }
+    
+    public int find(int number) {
+        TreeSet<Integer> indices = numberToIndex.getOrDefault(number, new TreeSet<>());
+        if (indices.isEmpty()) {
+            return -1;
+        }
+        return indices.first();
+    }
+}
+
+/**
+ * Your NumberContainers object will be instantiated and called as such:
+ * NumberContainers obj = new NumberContainers();
+ * obj.change(index,number);
+ * int param_2 = obj.find(number);
+ */
