原题指路
寻找数组的中心索引
题目描述
给你一个整数数组 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;
}
};