Merge pull request neetcode-gh#2221 from AkifhanIlgaz/0904

Create: 0904-fruit-into-baskets

Author: tahsintunan

go/0904-fruit-into-baskets.go

@@ -0,0 +1,34 @@
+package main
+
+func main() {
+
+}
+
+func totalFruit(fruits []int) int {
+	count := map[int]int{}
+	left, total, res := 0, 0, 0
+
+	for _, fruit := range fruits {
+		count[fruit]++
+		total++
+
+		for len(count) > 2 {
+			f := fruits[left]
+			count[f]--
+			total--
+			left++
+			if count[f] == 0 {
+				delete(count, f)
+			}
+		}
+		res = max(res, total)
+	}
+	return res
+}
+
+func max(a, b int) int {
+	if a > b {
+		return a
+	}
+	return b
+}

javascript/0904-fruit-into-baskets.js

@@ -0,0 +1,26 @@
+/**
+ * @param {number[]} fruits
+ * @return {number}
+ */
+var totalFruit = function (fruits) {
+    let count = new Map();
+    let [left, total, res] = [0, 0, 0];
+
+    for (fruit of fruits) {
+        count.set(fruit, (count.get(fruit) || 0) + 1);
+        total++;
+
+        while (count.size > 2) {
+            let f = fruits[left];
+            count.set(f, count.get(f) - 1);
+            total -= 1;
+            left += 1;
+            if (!count.get(f)) {
+                count.delete(f);
+            }
+        }
+        res = Math.max(res, total);
+    }
+
+    return res;
+};

rust/0904-fruit-into-baskets.rs

@@ -0,0 +1,29 @@
+use std::collections::HashMap;
+
+impl Solution {
+    pub fn total_fruit(fruits: Vec<i32>) -> i32 {
+        let mut count = HashMap::new();
+        let (mut left, mut total, mut res) = (0, 0, 0);
+
+        for fruit in &fruits {
+            *count.entry(*fruit).or_insert(0) += 1;
+            total += 1;
+
+            while count.len() > 2 {
+                let f = fruits[left];
+                match count.remove(&f) {
+                    Some(v) if v > 1 => {
+                        count.insert(f, v - 1);
+                    }
+                    _ => {}
+                }
+                total -= 1;
+                left += 1;
+            }
+
+            res = res.max(total);
+        }
+
+        res
+    }
+}

typescript/0904-fruit-into-baskets.ts

@@ -0,0 +1,23 @@
+function totalFruit(fruits: number[]): number {
+    let count = new Map();
+    let [left, total, res] = [0, 0, 0];
+
+    for (let fruit of fruits) {
+        count.set(fruit, (count.get(fruit) || 0) + 1);
+        total++;
+
+        while (count.size > 2) {
+            let f = fruits[left];
+            count.set(f, count.get(f) - 1);
+            total--;
+            left++;
+            if (!count.get(f)) {
+                count.delete(f);
+            }
+        }
+
+        res = Math.max(res, total);
+    }
+
+    return res;
+}