寻找数组的中心索引

最新推荐文章于 2022-01-07 11:09:58 发布

原创最新推荐文章于 2022-01-07 11:09:58 发布 · 255 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #Java #数组

本文介绍了一种算法，用于在给定的整数数组中找到一个中心索引，该索引左侧所有元素的和等于其右侧所有元素的和。若不存在这样的索引则返回-1。文章提供了Java实现代码，并通过示例展示了如何使用该算法。

题目

给定一个整数类型的数组 nums，请编写一个能够返回数组“中心索引”的方法。
我们是这样定义数组中心索引的：数组中心索引的左侧所有元素相加的和等于右侧所有元素>相加的和。
如果数组不存在中心索引，那么我们应该返回 -1。如果数组有多个中心索引，那么我们应该返回最靠近左边的那一个。

示例

示例 1:
输入:
nums = [1, 7, 3, 6, 5, 6]
输出: 3
解释:
索引3 (nums[3] = 6) 的左侧数之和(1 + 7 + 3 = 11)，与右侧数之和(5 + 6 = 11)相等。
同时, 3 也是第一个符合要求的中心索引。
示例 2:
输入:
nums = [1, 2, 3]
输出: -1
解释:
数组中不存在满足此条件的中心索引。
说明:

注意

nums 的长度范围为 [0, 10000]。
任何一个 nums[i] 将会是一个范围在 [-1000, 1000]的整数。

解

语言：Java

1.判断nums.length 是否为0；是，则返回0
2.leftSum表示左部分的和， sum表示右半部分的和；
3.从index：0处开始寻找：
如果满足，则返回：0；
否则，从index：1依次向后查询；
直至寻到 leftSum 等于 sum，或到达尾部

class Solution {
    public int pivotIndex(int[] nums) {
        //第一步：判断nums.length 是否为0；是，则返回0
        if(nums.length == 0) return -1;

        //sum表示右半部分的和
        double sum = 0;
        int i = 0;
        for(; i < nums.length; i++){
            sum += nums[i];
        }

        //从index：0查询；可以，则返回0；
        i = 0;
        long leftSum = 0;
        if(leftSum == (sum - nums[0])) return 0;

        //从index：2到index：nums.length - 1查询
        leftSum = nums[0];
        sum -= (nums[0] + nums[1]);
        while(((leftSum != sum) ||  (leftSum < sum)) && i < nums.length - 2){
            i ++;
            leftSum += nums[i];
            sum -= nums[i+1];
        }
        
       //假如index：nums.length - 1，leftSum 为0，则返回 nums.length - 1
       //原因：由于index：0——index：nums.length - 1，而前面的那一步index：nums.length - 1未判断
       if((i == nums.length - 1) && (leftSum + nums[nums.length - 2]) == 0) return nums.length - 1;

        //index 不为 nums.length - 1
        if(leftSum == sum) return i+1;
        else return -1;
    }
    
}

来源：力扣