LeetCode 724：寻找数组的中心索引 Find Pivot Index

爱写bug

发布于 2019-08-01 10:24:04

5330

发布于 2019-08-01 10:24:04

文章被收录于专栏：爱写Bug

Given an array of integers nums, write a method that returns the "pivot" index of this array.

We define the pivot index as the index where the sum of the numbers to the left of the index is equal to the sum of the numbers to the right of the index.

If no such index exists, we should return -1. If there are multiple pivot indexes, you should return the left-most pivot index.

给定一个整数类型的数组 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:

class Solution {
    public int pivotIndex(int[] nums) {
       int sumLeft=0,sum=0;
        for (int n:nums){
            sum = sum + n;
        }
        int numsSize=nums.length-1;
        for(int i=0;i<=numsSize;i++){
            if(i==0){
                sumLeft=0;
            }else{
                sumLeft+=nums[i-1];
            }
            //左侧索引如果和右侧相等，即二倍左侧加中心索引=总和
            if(sumLeft*2+nums[i]==sum){
                return i;
            }
        }
       return -1;
    }
}

特别说明一下，索引一定要从0开始到最后一个止，因为题目中心索引右侧可以为负数，相加可能为0。

python3

class Solution:
    def pivotIndex(self, nums: List[int]) -> int:
        """
        :type nums: int
        """
        sumAll=sum(nums)
        leftSum=0
        for i in range(len(nums)):
            if(i==0):
                leftSum=0
            else:
                leftSum+=nums[i-1]
            #左侧索引如果和右侧相等，即二倍加左侧中心索引=总和
            if(leftSum*2+nums[i]==sumAll):
                return i
        return -1

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2019-07-05，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法

本文分享自爱写Bug 微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

编程算法

登录后参与评论

0 条评论

热度