题目描述:
你一个整数数组 nums,请编写一个能够返回数组 “中心下标” 的方法。
数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果数组不存在中心下标,返回 -1 。如果数组有多个中心下标,应该返回最靠近左边的那一个。
注意:中心下标可能出现在数组的两端。
示例 1:
输入:nums = [1, 7, 3, 6, 5, 6]
输出:3
解释:
中心下标是 3 。
左侧数之和 (1 + 7 + 3 = 11),
右侧数之和 (5 + 6 = 11) ,二者相等。
示例 2:
输入:nums = [1, 2, 3]
输出:-1
解释:
数组中不存在满足此条件的中心下标。
示例 3:
输入:nums = [2, 1, -1]
输出:0
解释:
中心下标是 0 。
下标 0 左侧不存在元素,视作和为 0 ;
右侧数之和为 1 + (-1) = 0 ,二者相等。
算法思想:
开辟rightSum和leftSum两个变量,分别记录每个元素(0<=i<=nums.length-1)左面子数组所有元素的和与右面子数组所有元素的的和,比如数组为{1,2,3,4,5},当指针i取值为2(位置为2)的 时候,leftSum=1+2,rightSum=4+5,故先将整个数组元素进行累加,得到的结果赋值给rightSum,并且将leftSum设置为0,这样的情况下再次遍历整个数组,每当指针移动一下,rightSum减掉当前元素,leftSum加上当前元素,直到rightSum和leftSum相等,则返回当前的指针位置,就是我们寻找的数组的中心 下标,或者等到数组全部遍历结束,还没有找到rightSum和leftSum相等的位置,那样就要返回-1,如此就会得到题目的要求,其时间复杂度为O(n),空间复杂度为O(2).
代码实现:
public int pivotIndex(int[] nums) {
int j=0;
int rightSum=0,leftSum=0;
for(int i=0;i<nums.length;++i){
rightSum+=nums[i];
}
while(j<nums.length){
if(leftSum==rightSum-nums[j]){
return j;
}else{
leftSum+=nums[j];
rightSum-=nums[j];
j++;
}
}
return -1;
}