題目描述

題目分析

今天這道題原本很簡單，我都沒打算寫題解，當時用手機看的題目，我想著我三分鐘應該能寫出來，結果沒想到wa了三發。。。
對待簡單題不要輕視，對待難題不要畏難。
今天的主要問題是沒有看數據范圍，雖然只是一道很簡單的題，但是自從刷力扣以來自己基本不怎么關注數據范圍、邊界條件，總是等報錯以后才去看，這是一個很差勁的習慣，實際編程環境中是不會有現成的數據供測試的。
沒有看到題目中數組可能為空，而沒有想到中心索引可以是最左側或者最右側的。

AC代碼

class Solution {
public:int pivotIndex(vector<int>& nums) {int n = nums.size();if (n == 0) {return -1;}for (int i = 1; i < n; ++i) {nums[i] += nums[i-1];}if (nums[n-1] - nums[0] == 0) {return 0;}for (int i = 1; i < n-1; ++i) {if (nums[i-1] == nums[n-1] - nums[i]) {return i;}}if (nums[n-2] == 0) {return n-1;}return -1;}
};

官方題解

class Solution {
public:int pivotIndex(vector<int> &nums) {int total = accumulate(nums.begin(), nums.end(), 0);int sum = 0;for (int i = 0; i < nums.size(); ++i) {if (2 * sum + nums[i] == total) {return i;}sum += nums[i];}return -1;}
};//作者：LeetCode-Solution
//鏈接：https://leetcode-cn.com/problems/find-pivot-index/solution/xun-zhao-shu-zu-de-zhong-xin-suo-yin-by-gzjle/
//來源：力扣（LeetCode）
//著作權歸作者所有。商業轉載請聯系作者獲得授權，非商業轉載請注明出處。

感覺看官方的代碼每次都收益頗多
首先，對中心索引的條件進行了轉換，其次使用了accumulate函數進行求和
我的代碼寫出來感覺并不優美，那么多的特判。