数组列表的中心索引

求取数组列表的中心索引

题目：

给定一个整数类型的数组 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
解释: 数组中不存在满足此条件的中心索引。

代码实现

def pivotIndex(nums):
    """中心索引的左边和等于右边和,进行动态变化,右边和=nums总和-左边和-nums[i]"""
    if len(nums) == 1: return 0  # 如果长度为1,直接返回0索引
    if len(nums) == 0: return -1  # 如果nums为空,则直接返回-1
    i = 0
    left_sum = 0
    sums = sum(nums)  # 计算总和
    while i < len(nums):
        if left_sum == (sums - left_sum - nums[i]):
            return i
        left_sum += nums[i]
        i += 1
    return -1
    
#测试
nums = [1, 7, 3, 6, 5, 6]
index = pivotIndex(nums)
print(index) #结果为3

总结

第一次实现错误：

思想: 从左右两边同时计数,如果左边和大于右边和，则右边加下一个数，若右边大则左边加下一个数，直到两边和相等结束。

错误在于左边和或者右边和可能会越加越小，比如nums为[-1,-1,-1,-1,-1,0]