LeetCode简单题之寻找数组的中心下标
題目
給你一個整數數組 nums ,請計算數組的 中心下標 。
數組 中心下標 是數組的一個下標,其左側所有元素相加的和等于右側所有元素相加的和。
如果中心下標位于數組最左端,那么左側數之和視為 0 ,因為在下標的左側不存在元素。這一點對于中心下標位于數組最右端同樣適用。
如果數組有多個中心下標,應該返回 最靠近左邊 的那一個。如果數組不存在中心下標,返回 -1 。
示例 1:
輸入:nums = [1, 7, 3, 6, 5, 6]
輸出:3
解釋:
中心下標是 3 。
左側數之和 sum = nums[0] + nums[1] + nums[2] = 1 + 7 + 3 = 11 ,
右側數之和 sum = nums[4] + nums[5] = 5 + 6 = 11 ,二者相等。
示例 2:
輸入:nums = [1, 2, 3]
輸出:-1
解釋:
數組中不存在滿足此條件的中心下標。
示例 3:
輸入:nums = [2, 1, -1]
輸出:0
解釋:
中心下標是 0 。
左側數之和 sum = 0 ,(下標 0 左側不存在元素),
右側數之和 sum = nums[1] + nums[2] = 1 + -1 = 0 。
提示:
1 <= nums.length <= 10^4
-1000 <= nums[i] <= 1000
來源:力扣(LeetCode)
解題思路
??對于這樣比較直白的題可以先直接用代碼來翻譯題目,然后再看看是否能夠優化。
class Solution:def pivotIndex(self, nums: List[int]) -> int:for i in range(len(nums)):if sum(nums[0:i])==sum(nums[i+1:]):return ireturn -1
??通過以上的代碼,不難發現,其實每次計算sum的時候我們總要重復計算前面已經算過的值,這樣就比較浪費算力。數組中某一個元素兩側的和相等,也就是當我們減掉這個元素然后將整個數組的和除以二那么就是一側的和,這樣做至少會減少一半的多余計算,盡管我們還有一半的計算是重復的。
class Solution:def pivotIndex(self, nums: List[int]) -> int:total=sum(nums) #總和s=0 #從左邊累加for i in range(len(nums)):if 2*s+nums[i]==total: #如果左側的值的二倍加上中間的樞軸值等于總和則找到中間位置return is+=nums[i]return -1
總結
以上是生活随笔為你收集整理的LeetCode简单题之寻找数组的中心下标的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode简单题之棒球比赛
- 下一篇: LeetCode简单题之词典中最长的单词