题目描述
题目分析
今天这道题原本很简单,我都没打算写题解,当时用手机看的题目,我想着我三分钟应该能写出来,结果没想到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
函数进行求和
我的代码写出来感觉并不优美,那么多的特判。