add 0904-fruit-into-baskets.py

Author: Dhyan-P-Shetty

python/0904-fruit-into-baskets.py

@@ -0,0 +1,29 @@
+import collections
+
+class Solution:
+    def totalFruit(self, fruits: List[int]) -> int:
+        count = collections.defaultdict(int)
+        l, total, res = 0, 0, 0
+
+        for r in range(len(fruits)):
+            count[fruits[r]] += 1
+            total += 1
+
+            while len(count) > 2:
+                f = fruits[l]
+                count[f] -= 1
+                total -= 1
+                l += 1
+                if not count[f]:
+                    count.pop(f)
+
+            res = max(res, total)
+
+        return res
+
+
+        
+            
+
+            
+