给你一个整数数组 nums
,请计算数组的 中心下标 。
数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果中心下标位于数组最左端,那么左侧数之和视为 0
,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1
。
链接:724. 寻找数组的中心下标 - 力扣(LeetCode)
方法一:
class Solution {
public:
int pivotIndex(vector<int>& nums) {
int n = nums.size();
int sum = 0;
for (int i = 0; i < n; i++) {
sum += nums[i]; // 计算数组总和
}
int leftSum = 0;
for (int i = 0; i < n; i++) {
sum -= nums[i]; // 减去当前元素,得到右侧元素之和
if (leftSum == sum) {
return i; // 如果左侧元素之和等于右侧元素之和,返回当前索引
}
leftSum += nums[i]; // 累加当前元素到左侧元素之和
}
return -1; // 如果找不到枢轴索引,返回-1
}
};
开始打算同时两边开始加每循环一次判断一次,判断左右是否相等,十分麻烦。最后选择另一种思路,先求合,在通过减法(减去当前值) 得到右边的值,在判断,比第一种要考虑的情况少很多。