problems: add find pivot index
This commit is contained in:
parent
b100c7588b
commit
af0dd6af82
1 changed files with 20 additions and 0 deletions
20
problems/find-pivot-index.rs
Normal file
20
problems/find-pivot-index.rs
Normal 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
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue