Merge pull request #3158 from a1exanddrovich/0904-dev

Create: 0904-fruit-into-baskets.java

Author: a93a

java/0904-fruit-into-baskets.java

@@ -0,0 +1,25 @@
+class Solution {
+    public int totalFruit(int[] fruits) {
+        Map<Integer, Integer> fruitTypeToItsCount = new HashMap<>();
+        int maxTotalFruits = 0;
+        int l = 0;
+        int r = 0;
+
+        while (r < fruits.length) {
+            fruitTypeToItsCount.put(fruits[r], fruitTypeToItsCount.getOrDefault(fruits[r], 0) + 1);
+
+            while (fruitTypeToItsCount.size() > 2) {
+                fruitTypeToItsCount.put(fruits[l], fruitTypeToItsCount.get(fruits[l]) - 1);
+                if (fruitTypeToItsCount.get(fruits[l]) == 0) {
+                    fruitTypeToItsCount.remove(fruits[l]);
+                }
+                l++;
+            }
+
+            maxTotalFruits = Math.max(maxTotalFruits, r - l + 1);
+            r++;
+        }
+
+        return maxTotalFruits;
+    }
+}