3100. 换水问题 II

src/main.rs

@@ -2,17 +2,18 @@ mod arr;
 struct Solution {}
 
 impl Solution {
-    pub fn triangular_sum(mut nums: Vec<i32>) -> i32 {
-        for i in 0.. nums.len() - 1 {
-            for j in 0..nums.len() - 1 - i {
-                nums[j] = (nums[j] + nums[j + 1]) % 10;
-            }
+    pub fn max_bottles_drunk(mut num_bottles: i32,mut num_exchange: i32) -> i32 {
+        let mut ans = num_bottles;
+        while num_bottles >= num_exchange {
+            num_bottles -= num_exchange;
+            ans += 1;
+            num_bottles += 1;
+            num_exchange += 1;
         }
-        nums[0]
+        ans
     }
 }
 
 fn main() {
-    let arr = vec![1,2,3,4,5];
-    println!("{}", Solution::triangular_sum(arr));
+    println!("{}", Solution::max_bottles_drunk(10,3));
 }