给你一个整数数组 nums ,请编写一个能够返回数组"中心索引"的方法。数组中心索引是数组的一个索引,其左侧所有元素相加的和等于右侧所有元素相加的和。如果数组不存在中心索引,返回-1如果数组有多个中心索引,应该返回最靠近左边的那一个。注意:中心索引可能出现在数组的两端。
这道题的思路比较简单,就是计算前i个数的和是否等于i后面的数的和。也是第一次接触前缀和的概念,如果没接触这个概念那我写这道题的方法就是用两个int类型的变量分别来存储i以前和i以后的值,然后比较:
import java.util.Scanner;
public class Text01 {
public static void main(String[] args) {
int[] arr = new int[new Scanner(System.in).nextInt()];
String[] arr_s = new Scanner(System.in).nextLine().split(" ");
for(int i=0; i<arr_s.length; i++) {
arr[i] = Integer.parseInt(arr_s[i]);
}
System.out.println(middle(arr));
}
public static int middle(int[] arr) {
for(int i=0; i<arr.length; i++) {
int sum1 = 0;
int sum2 = 0;
if(i==0 || i==arr.length-1) {
for(int j=0; j<arr.length; j++) {
sum2 += arr[j];
}
if(sum2 == sum1) {
return i;
}
}else {
for(int j=0; j<arr.length; j++) {
if(j<i) {
sum1 += arr[j];
}else if(j>i){
sum2 += arr[j];
}
}
if(sum2 == sum1) {
return i;
}
}
}
return -1;
}
}
对于前缀和的计算就是对于给定的数组arr1,新建一个数组arr2,arr2[i] = arr1[0]+arr1[1]+…+arr1[i];即arr2[i]存储的是arr1的前i项和。这样计算就显得方便很多:
import java.util.Scanner;
public class Text01 {
public static void main(String[] args) {
int[] arr = new int[new Scanner(System.in).nextInt()];
String[] arr_s = new Scanner(System.in).nextLine().split(" ");
for(int i=0; i<arr_s.length; i++) {
arr[i] = Integer.parseInt(arr_s[i]);
}
System.out.println(middle(arr));
}
public static int middle(int[] arr) {
int[] arr1 = new int[arr.length];
arr1[0] = arr[0];
for(int i=1; i<arr1.length; i++) {
arr1[i] = arr1[i-1]+arr[i];
}
for(int i=0; i<arr1.length; i++) {
if(arr1[i]-arr[i] == arr1[arr1.length-1]-arr1[i]) {
return i;
}
}
return -1;
}
}