- 链接
https://leetcode.cn/problems/tvdfij/ - 题目
给你一个整数数组 nums ,请计算数组的 中心下标 。
数组中心下标是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果中心下标位于数组最左端,那么左侧数之和视为 0 ,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1 。
示例:
输入:nums = [1,7,3,6,5,6]
输出:3
解释:
中心下标是 3 。
左侧数之和 sum = nums[0] + nums[1] + nums[2] = 1 + 7 + 3 = 11 ,
右侧数之和 sum = nums[4] + nums[5] = 5 + 6 = 11 ,二者相等。
-
思路:
- 先求出数组的总和
sum
; - 再进行遍历,并计算当前下标
i
左边子数组的和,记为lsum
; - 设未知数右边子数组之和为
rsum
。根据题目意思,rsum=sum-lsum-nums[i]
,如果当前i
就是中心下标,那么rsum=lsum
,即sum=2*lsum+nums[i]
。
- 先求出数组的总和
-
代码:
class Solution {
public:
int pivotIndex(vector<int>& nums) {
int sum=0,lsum=0,k;
for(int i=0;i<nums.size();i++)
sum+=nums[i];
for(k=0;k<nums.size();k++){
if(2*lsum+nums[k]==sum)
return k;
lsum+=nums[k];
}
return -1;
}
};