[閒聊] 每日leetcode 75 - Day8 - 3

[yam276 (史萊哲林的優等生)]

724. Find Pivot Index


題目：

找到陣列中的一個甜蜜點

甜蜜點左邊總和 = 甜蜜點右邊總和


思路：

跑兩次迴圈

第一次求總和

第二次則每次先判斷 左邊總和 跟 右邊總和-甜蜜點

之後更新數值

因為甜蜜點不算在內


Code:

impl Solution {
    pub fn pivot_index(nums: Vec<i32>) -> i32 {
        let mut right_sum: i32 = nums.iter().sum();
        let mut left_sum = 0;

        for (i, num) in nums.iter().enumerate() {
            if left_sum == right_sum - num {
                return i as i32;
            }

            left_sum += num;
            right_sum -= num;
        }
        -1
    }
}

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.32.48.170 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1749539468.A.D1F.html
→ oin1104: 大師