【LeetCode 之数组和字符串】寻找数组的中心索引

【LeetCode 之数组和字符串】寻找数组的中心索引

    题目描述：（ 题目链接：https://leetcode-cn.com/leetbook/read/array-and-string/yf47s/ ）

给你一个整数数组 nums，请编写一个能够返回数组 “中心下标” 的方法。

数组 中心下标 是数组的一个下标，其左侧所有元素相加的和等于右侧所有元素相加的和。

如果数组不存在中心下标，返回 -1 。如果数组有多个中心下标，应该返回最靠近左边的那一个。

注意：中心下标可能出现在数组的两端。

示例 1：

输入：nums = [1, 7, 3, 6, 5, 6]
输出：3
解释：
中心下标是 3 。
左侧数之和 (1 + 7 + 3 = 11)，
右侧数之和 (5 + 6 = 11) ，二者相等。


示例 2：

输入：nums = [1, 2, 3]
输出：-1
解释：
数组中不存在满足此条件的中心下标。


示例 3：

输入：nums = [2, 1, -1]
输出：0
解释：
中心下标是 0 。
下标 0 左侧不存在元素，视作和为 0 ；
右侧数之和为 1 + (-1) = 0 ，二者相等。


提示：nums 的长度范围为 [0, 10000]。
     任何一个 nums[i] 将会是一个范围在 [-1000, 1000]的整数。

 

思路：求中心索引，条件是 nums[中心索引] 的左侧数据元素之和 = 右侧数据元素之和。如果不满足这个条件，则返回 -1。

也就是对数组里的元素进行遍历操作，特别要注意的是从 nums[0] 开始，它的左侧元素之和为 0，也就是示例 2。（这也是为何 rightSum 的值是总和减掉数组里的第一个元素）

而判断 “左和 = 右和” 条件，楼主先定义左侧元素之和 leftSum 为0，并计算了数组里所有元素的总和，用的是 accumulate() 函数，很方便（因为题目给的是用 vector，所以先直接找其相关函数学习 ），如果相等的话，直接返回下标就行。

不相等的话，就往后遍历。这时候索引从数组下一个地址移动，所以 “右和” 要减去这下一个数值，“左和” 则加上去。

（楼主看到题目的评论区有人讨论数组为空或是为 0 的情况要加上去，在这里，加不加都通过了，只不过时间上有点点延迟。）

// 第一种：加了判断 nums 是否为空或为0
class Solution {
public:
    int pivotIndex(vector<int>& nums) {
         if (nums.empty() || nums.size() == 0)
            return -1;
        int leftSum = 0;
        int totalSum = accumulate(nums.begin(), nums.end(), 0);
        int rightSum = totalSum - nums[0];
        for (int i = 0; i < nums.size(); i++) {
            if (leftSum == rightSum) {
                return i;
            }
            rightSum -= nums[i + 1];
            leftSum += nums[i];
        }
        return -1;
    }
};


// 第二种：未加判断
class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int leftSum = 0;
        int totalSum = accumulate(nums.begin(), nums.end(), 0);
        int rightSum = totalSum - nums[0];
        for (int i = 0; i < nums.size(); i++) {
            if (leftSum == rightSum) {
                return i;
            }
            rightSum -= nums[i + 1];
            leftSum += nums[i];
        }
        return -1;
    }
};

下面是官方解：（这个也好理解，官方说明得很清楚，只不过楼主想的是，数学很重要，不要问，问多了都是刚成为考研党的辛酸。）

（给大家直接解释吧：在第 i 个元素上，左和 = 右和，又 右和 = 总和 - 左和 - nums[i]，所以 2 * 左和 + nums[i] = 总和）

ps:  为更好地理解，可以把数组的第一个元素代进去想上面讲的式子。

class Solution {
public:
    int pivotIndex(vector<int> &nums) {
        int total = accumulate(nums.begin(), nums.end(), 0);
        int leftSum = 0;
        for (int i = 0; i < nums.size(); ++i) {
            if (2 * leftSum + nums[i] == total) {
                return i;
            }
            leftSum += nums[i];
        }
        return -1;
    }
};

其实，楼主想的，和官方解本质是一样的，就多了个 “右和” 的定义。但楼主还是感觉官方还就是官方，看起来更高大上。

--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

楼主有话要说：

现在是用 LeetCode 来练习算法的第二天。虽然是个小白，也想记录下思路，也便于之后的复习。

楼主说话直白，本人对于计算机也算是个小白，因为啥也不太懂，所以，在做题之前，在 LeetCode 中学习进行学习专项课程，先从数组开始学起。 其中，不仅看《数据结构（C语言版）（第2版）》教材，还在网上查询相关的资料。搜到了一些很不错的学习内容，分享（目前主要用的就这三个）：

1、LeetCode 的《数组和字符串》课程   https://leetcode-cn.com/leetbook/detail/array-and-string/

ps: 一边学知识点，一边刷题。不过楼主刷得慢，就做了三道题，这三道题都有官方解还有众多大佬的思路和其它解法，楼主感到满满的幸福感！

2、刷leetcode不知道从哪里刷起？？ 来看看，我把刷题题目顺序都排好了，每道题还带详细题解！  https://blog.csdn.net/youngyangyang04/article/details/111773608

ps: 这篇是真的不错，点进去之后是跳转到大佬在微信公众号里分享的学习内容。对于代码风格，学并用的是小驼峰（力扣在用），看起来确实舒服。楼主是搭配着 LeetCode 的《数组和字符串》课程来学的。收获满满，适合小白。

3、牛客网的“专项练习”。https://www.nowcoder.com/intelligentTest

ps:  因为在学数组，所以找了相应的“数据结构”里的数组类型题来做，巩固，也相当于复习了。接下来，牛客网的专项习题里，有时间也会刷。

 

不得不说的是，因为是小白，vector 相关的知识，比如 find(), accumulate() 函数都得重新学，所以又得查，考研这条路难走，但楼主已经没有退路了，只能向前冲，加油。