数组与字符串——寻找数组的中心索引
程序员文章站
2022-03-21 17:26:19
...
没时间了,先上题!
给定一个整数类型的数组 nums,请编写一个能够返回数组 “中心索引” 的方法。
我们是这样定义数组 中心索引 的:数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。如果数组不存在中心索引,那么我们应该返回 -1。如果数组有多个中心索引,那么我们应该返回最靠近左边的那一个。
解题思路
-
最先想到既然题目已经给出了数组的总长度,那么就可以定义两个整型i和j,分别从数组两端开始遍历,左边i遍历过的总和为leftSum,右边j遍历过的总和为rightSum。如果leftSum>rightSum就执行j–,如果leftSum<rightSum就执行i++,否则i–与j++同时进行,直到左右两边和相等且满足j-i=2。
错误原因:没有考虑到和越加越小的情况,即下一个要加的数可能为负数。 -
正确思路:只需要首先遍历完整个数组求出总和sum,然后逐个从j=0开始寻找满足leftSum=sum-leftSum-nums[j],满足即返回中心索引 j(这样也满足了存在多个中心索引时首先返回的是最左边的一个)否则遍历完毕后仍未找到中心索引则返回-1。
代码
int pivotIndex(int* nums, int numsSize){
int sum = 0;
int leftSum = 0;
//求数组和
for(int i = 0; i < numsSize; i++){
sum += nums[i];
}
//从头依次寻找中心索引
for(int j = 0; j < numsSize; j++){
//当前索引为中心索引满足的判断表达式
if(leftSum == sum-nums[j]-leftSum){
return j;
}
leftSum += nums[j];
}
return -1;
}