724. Find Pivot Index
724. Find Pivot Index
這一題的目標是我們要找到在陣列中的一個索引,這個索引左側的所有數字的總和相等於右側的所有數字總和,索引的位置可能在這個陣列中的任何一個地方,取決於左側和右側每個數字的大小以及其分佈。
這一題比較直觀的想法是,那我先計算出這個陣列從左到右的累加總和,再計算出這個陣列從右到左的累加總和,這兩個陣列中如果有同一個位置剛好數字一樣,那就是我們要找的索引。
題目給的範例:
nums = [1,7,3,6,5,6]
accumulate_sum_from_left = [ 1, 8,11,17,22,28]
accumulate_sum_from_right = [28,27,20,17,11, 6]
如果我們先計算出上面兩個數值,就可以很快地找到位置是在 3 。下面是程式的實作:
class Solution:
def pivotIndex(self,