problems: add find pivot index

This commit is contained in:
Matej Focko 2022-07-12 12:36:28 +00:00
parent b100c7588b
commit af0dd6af82

View file

@ -0,0 +1,20 @@
use std::convert::TryInto;
impl Solution {
pub fn pivot_index(nums: Vec<i32>) -> i32 {
let mut from_left: i32 = 0;
let mut from_right: i32 = nums.iter().sum();
for (i, e) in nums.iter().enumerate() {
from_right -= e;
if from_left == from_right {
return i.try_into().unwrap();
}
from_left += e;
}
-1
}
}