寻找数组的中心索引

原题指路
寻找数组的中心索引

题目描述

给你一个整数数组 nums，请编写一个能够返回数组 “中心索引” 的方法。
数组 中心索引 是数组的一个索引，其左侧所有元素相加的和等于右侧所有元素相加的和。

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

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

解题思路

1、首先求出数组长度，若长度为0，可直接返回-1，即找不到；

2、接着调用STL库函数accumulate函数求出从数组第2个元素到数组最后一个元素的元素和，记为total；

3、若此时total=0，则第1个元素即为数组的中心索引，直接返回0；

4、接下来从数组第2个元素开始执行循环，直到找到中心索引或循环至数组最后一个元素；

5、若循环结束仍未找到，则说明该数组无中心索引，返回-1。

时间复杂度分析：O（n）

空间复杂度分析：O（1）

AC代码

class Solution 
{
    public:
        int pivotIndex(vector<int>& nums) 
        {
            int sum=0,length=nums.size();
            if(!length)//数组长度为0，直接返回-1
                return -1;
            int total=accumulate(nums.begin()+1,nums.end(),0);
            if(!total)//数组第1个元素记为中心索引
                return 0;
            for(int i=1;i<length;i++)
            {
                sum+=nums[i-1];
                total-=nums[i];
                if(sum==total)//找到中心索引，直接返回当前索引
                    return i;
            }
            return -1;
        }
};