diff --git a/find_pivot_index/src/main.rs b/find_pivot_index/src/main.rs index 5463c27..e024fb2 100644 --- a/find_pivot_index/src/main.rs +++ b/find_pivot_index/src/main.rs @@ -3,7 +3,6 @@ fn main() { } fn f(nums: Vec) -> i32 { - let mut pivot = -1; let mut rights = Vec::with_capacity(nums.len()); let mut cur = 0; for n in nums.iter().rev() { @@ -16,9 +15,8 @@ fn f(nums: Vec) -> i32 { for (i, num) in nums.into_iter().enumerate() { cur += num; if cur == rights[i] { - pivot = i as i32; - break; + return i as i32; } } - pivot + -1 }