题目:给定一个整数类型的数组nums,请编写一个能够返回数组“中心索引”的方法。
我们是这样定义数组中心索引的:数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。如果数组不存在中心索引,那么我们应该返回-1.如果数组有多个中心索引,那么我们应该返回最靠近左边的那一个。
示例1:
输入:nums = [1,7,3,6,5,6]
输出:3
解释:索引3(nums[3] = 6)的左侧数之和(1+7+3 = 11),与右侧数之和(5+6 = 11)相等。同时,3也是第一个符合要求的中心索引。
示例2:
输入:nums = [1,2,3]
输出:-1
解释:数组中不存在满足条件的中心索引。
说明:
nums的长度范围为[0,10000]。
任何一个nums[i]将会是一个范围在[-1000,1000]的整数。
实现思路:根据中心索引的定义,从数组左侧开始加,设左侧值为sum,那么满足条件的右侧值为total-sum-nums[i],即当二者相等时返回索引i;简化之后即2*sum+nums[i] == total(因为左侧等于右侧所有元素相加的和)。
C++实现:
class Solution {
public:
int pivotIndex(vector<int>& nums) {
int total = accumulate(nums.begin(), nums.end(), 0); //total等于nums数组元素之和
int sum = 0;
for (int i = 0; i < nums.size(); ++i) {
if (2 * sum + nums[i] == total) { //如果左边+右边(即2倍的左边)+小标为i的元素等于total,返回索引
return i;
}
sum += nums[i]; //sum等于左侧数之和
}
return -1; //不满足条件返回-1
}
};
结果: