给定一个整数数组nums,请编写一个能够返回数组“中心下标”的方法
中心下标是数组的一个下标,其左侧的所有元素相加的合等于右侧所有元素相加的和。
如果数组不存在中心下标,返回-1。如果数组又多个中心下标,应该返回最靠近左边的哪一个。
注意中心下标可能出现在数组的两端。
import java.util.Arrays;
public class ArrayCenterIndex {
public static void main(String[] args) {
System.out.println("下标为:"+pivotIndex(new int[]{1,7,2,6,4,6}));
}
public static int pivotIndex(int[] nums){
//jdk1.8的新特性stream
int sum= Arrays.stream(nums).sum();
int total=0;
for (int i = 0; i < nums.length; i++) {
total+=nums[i];
//第一步先进行if对比是为了防止出现在两端的情况
if (total==sum) {
return i;
}
sum=sum-nums[i];
}
return -1;
}
}
可能第一眼不太理解为什么下标为3的时候左边等于右边,中心下标的值多少并不影响结果。
不妨我们设中心下标value=x,当中心下标出现时,左边的值=右边的值,x+左边的值=x+右边的值,所以中心下标的值不影响结果。