724. Find Pivot Index

最新推荐文章于 2021-04-26 10:57:58 发布

xiabe

最新推荐文章于 2021-04-26 10:57:58 发布

阅读量147

点赞数

分类专栏： leetcode 文章标签： python leetcode

本文链接：https://blog.csdn.net/xiabenshu/article/details/88870046

版权

leetcode 专栏收录该内容

75 篇文章 1 订阅

订阅专栏

题目

找到一个中间点，使左边的集合和右边集合的求和综合相等。
**注意：**集合可以为空。并且含正负数。
在这里插入图片描述

我的代码（效率跟优秀代码接近）

class Solution(object):
    def pivotIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums)<2:
            return -1
        left=0
        right=sum(nums[1:])
        for i in range(0,len(nums)):
            if right==left:
                return i
            else:
                left+=nums[i]
                if i+1<len(nums):
                    right-=nums[i+1]
        return -1

优秀代码（效率不错）

class Solution(object):
    def pivotIndex(self, nums):
        # Time: O(n)
        # Space: O(1)
        left, right = 0, sum(nums)
        for index, num in enumerate(nums):
            right -= num
            if left == right:
                return index
            left += num
        return -1