20240913 finished.

src/problem/mod.rs

@@ -233,4 +233,5 @@ mod p977_squares_of_a_sorted_array;
 mod p2181_merge_nodes_in_between_zeros;
 mod p2552_count_increasing_quadruplets;
 mod p2555_maximize_win_from_two_segments;
-mod p2576_find_the_maximum_number_of_marked_indices;
+mod p2576_find_the_maximum_number_of_marked_indices;
+mod p2398_maximum_number_of_robots_within_budget;

src/problem/p2398_maximum_number_of_robots_within_budget.rs

@@ -0,0 +1,66 @@
+/**
+ * [2398] Maximum Number of Robots Within Budget
+ */
+pub struct Solution {}
+
+
+// submission codes start here
+
+impl Solution {
+    pub fn maximum_robots(charge_times: Vec<i32>, running_costs: Vec<i32>, budget: i64) -> i32 {
+        use std::collections::VecDeque;
+
+        let n = charge_times.len();
+        let mut result = 0;
+        let mut running_costs_sum = 0i64;
+        let mut queue = VecDeque::with_capacity(n);
+
+        // 最左侧的机器人
+        let mut left = 0;
+
+        for i in 0..n {
+            running_costs_sum += running_costs[i] as i64;
+
+            // 维持单调队列
+            while let Some(&pos) = queue.back() {
+                if charge_times[pos] <= charge_times[i] {
+                    queue.pop_back();
+                } else {
+                    break;
+                }
+            }
+
+            queue.push_back(i);
+
+            // 判断当前left开始的机器人是否会导致超过预算
+            while left <= i && (i - left + 1) as i64 * running_costs_sum + charge_times[*queue.front().unwrap()] as i64 > budget {
+                if let Some(&front) = queue.front() {
+                    if front == left {
+                        queue.pop_front();
+                    }
+                }
+
+                running_costs_sum -= running_costs[left] as i64;
+                left += 1;
+            }
+            if i >= left {
+                result = result.max(i - left + 1);
+            }
+        }
+
+        result as i32
+    }
+}
+
+// submission codes end
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_2398() {
+        assert_eq!(3, Solution::maximum_robots(vec![3, 6, 1, 3, 4], vec![2, 1, 3, 4, 5], 25));
+        assert_eq!(0, Solution::maximum_robots(vec![11, 12, 19], vec![10, 8, 7], 19));
+    }
+}