题目要求:给定一个整数数组nums,编写返回数组中心下标的算法。
中心下标其左侧所有元素相加的和等于右侧所有元素相加的和。如果不存在,返回-1,如果存在多个,返回最左边的中心下标。
想法
使用双指针算法。首先计算出整个数组元素的和sum,之后设定i由数组左边开始遍历,每指向一个元素时,i代表的值增加,sum代表的值减少,直到两个的值相同为止。pivotIndexTest()是另一种实现方法,将判断条件改了,总体思路与上述一致。
代码实现
package test;
import java.util.Arrays;
public class ArrayCenterIndex {
public static void main(String[] args) {
System.out.println(pivotIndexTest(new int[] {1,7,3,6,5,6}));
}
public static int pivotIndex(int nums[]) {
int sum = Arrays.stream(nums).sum();
int total = 0;
for(int i = 0; i < nums.length;i++) {
total += nums[i];
if(total == sum) {
return i;
}
sum -= nums[i];
}
return -1;
}
public static int pivotIndexTest(int nums[]) {
int sum = Arrays.stream(nums).sum();
System.out.println(sum);
int total = 0;
for(int i = 0; i < nums.length;i++) {
total += nums[i];
//将数组分为两半,中心下标处左边值的二倍加上中心下标值一定等于sum
if(2 * total + nums[i+1] == sum) {
return i+1;
}
}
return -1;
}
}
总结
这个问题有很多种实现方法,本文写出来的两个方法是笔者的思路,欢迎大家指出新思路。