LeetCode刷题——Array篇(Find Pivot Index)

最新推荐文章于 2024-01-03 19:33:05 发布

lpcarl

最新推荐文章于 2024-01-03 19:33:05 发布

阅读量253

点赞数

CC 4.0 BY-SA版权

分类专栏： Array

本文链接：https://blog.csdn.net/lpcarl/article/details/80047902

Array 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种高效的方法来确定一个整数数组中的“中心”索引，即该位置左边所有元素之和等于右边所有元素之和的位置。通过一个具体的示例详细解释了算法的实现过程，并提供了一个简洁的C++代码示例。

Given an array of integers nums, write a method that returns the “pivot” index of this array.

We define the pivot index as the index where the sum of the numbers to the left of the index is equal to the sum of the numbers to the right of the index.

If no such index exists, we should return -1. If there are multiple pivot indexes, you should return the left-most pivot index.

Example 1:
Input:
nums = [1, 7, 3, 6, 5, 6]
Output: 3
Explanation:
The sum of the numbers to the left of index 3 (nums[3] = 6) is equal to the sum of numbers to the right of index 3.
Also, 3 is the first index where this occurs.
Example 2:
Input:
nums = [1, 2, 3]
Output: -1
Explanation:
There is no index that satisfies the conditions in the problem statement.

受前面题目影响

1、一开始想使用左右两个指针逐渐向中间靠拢的方式，但是遇到负数、指针回退的操作起来更加不方便；（后来想想应该是想错了，应该是一开始就该让指针靠在一起，然后一起移动）
2、看了答案，思路差不多，它是先求出所有数的和，然后逐一移动，前面的和与后面和之间的和是否一致

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