寻找数组的中心索引(Java求解)

题目:

给定一个整数类型的数组 nums,请编写一个能够返回数组“中心索引”的方法。

我们是这样定义数组中心索引的:数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。

如果数组不存在中心索引,那么我们应该返回 -1。如果数组有多个中心索引,那么我们应该返回最靠近左边的那一个。
 

暴力解法:

class Solution {
    public int pivotIndex(int[] nums) {
        for(int i=0;i<nums.length;i++){
            int sum1=0;
            int sum2=0;
            //计算左侧所有元素的和
            for(int j=i-1;j>=0;j--){
                sum1+=nums[j];
            }
            //计算右侧所有元素的和
            for(int j=i+1;j<nums.length;j++){
                sum2+=nums[j];
            }
            //判断左侧和与右侧和是否相等,相等则返回i值,循环结束
            if(sum1==sum2){
                return i; 
            }
        }
        return -1; //若无符合条件的中心索引,则在数组遍历结束后返回-1
    }
}

优化:

1.sumLeft + sumRight + nums[p] = sumTotal;
2.sumLeft = sumRight;
3.可以得出 sumLeft * 2 + nums[p] = sumTotal;

class Solution {
    public int pivotIndex(int[] nums) {
        int sumTotal = 0;
        int sumLeft = 0;
        for (int x: nums){
            sumTotal += x;
        }
        for (int i = 0; i < nums.length; i++) {
            if (sumLeft*2 == sumTotal -  nums[i]){
                 return i;
            }
            sumLeft += nums[i];
        }
        return -1;
    }
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-pivot-index

猜你喜欢

转载自blog.csdn.net/qq_41904729/article/details/102985406