Day6 中心索引

'''
给定一个整数类型的数组 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
解释: 数组中不存在满足此条件的中心索引。
请补充完成下面的代码：

'''


class Solution:
    # def pivotIndex(self, nums: List[int]) -> int:
    def pivotIndex(self, nums) -> int:
        if len(nums) <3:return -1
        left = 0
        right = len(nums) -1
        left_sum = nums[0]
        right_sum = nums[-1]
        while left + 1 != right:
            if left_sum == right_sum and left+2 == right:
                return left+1
            elif left_sum > right_sum:
                right -=1
                right_sum += nums[right]
            else:
                left +=1
                left_sum += nums[left]
        return -1

# s = Solution()
# print(s.pivotIndex([1,3,1]))




def test_pivotindex():
    s  = Solution()
    assert s.pivotIndex([1, 7]) == -1
    assert s.pivotIndex([1, 3, 1]) == 1
    assert s.pivotIndex([1, 7, 3, 6, 5, 6]) ==  3
    assert s.pivotIndex([1, 2, 3]) == -1