欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

数组与字符串——寻找数组的中心索引

程序员文章站 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;
}