Added Advantage Shuffle.java

varunu28 · web-flow · commit b2072a495a77 · 2021-03-24T07:07:26.000-07:00
diff --git a/Medium/Advantage Shuffle.java b/Medium/Advantage Shuffle.java
@@ -0,0 +1,31 @@
+class Solution {
+  public int[] advantageCount(int[] A, int[] B) {
+    TreeMap<Integer, Integer> map = new TreeMap<>();
+    for (int num : A) {
+      map.put(num, map.getOrDefault(num, 0) + 1);
+    }
+    int[] result = new int[A.length];
+    Arrays.fill(result, Integer.MIN_VALUE);
+    List<Integer> indexesNotPopulated = new ArrayList<>();
+    for (int i = 0; i < B.length; i++) {
+      Integer upper = map.higherKey(B[i]);
+      if (upper != null) {
+        result[i] = upper;
+        map.put(upper, map.get(upper) - 1);
+        if (map.get(upper) == 0) {
+          map.remove(upper);
+        }
+      } else {
+        indexesNotPopulated.add(i);
+      }
+    }
+    Iterator<Integer> iterator = indexesNotPopulated.iterator();
+    for (Integer key : map.keySet()) {
+      int value = map.get(key);
+      while (value-- > 0) {
+        result[iterator.next()] = key;
+      }
+    }
+    return result;
+  }
+}
