题解 | #相等的草堆#

相等的草堆

https://www.nowcoder.com/practice/0e2f3b27bbdc45fcbc70cc4fd41e15fe

  • 题目考察的知识点 : 数组模拟
  • 题目解答方法的文字分析:
  1. 先计算出所有草堆的总重量 sum,然后从左往右依次遍历每个草堆。在遍历每个草堆时,我们可以使用一个变量 left_sum 来记录它左边所有草堆的总重量,并用 sum 减去当前草堆的重量和 left_sum 得到它右边所有草堆的总重量。如果 left_sum 和 right_sum 相等,则说明当前草堆符合条件,可以返回索引值;否则,将 left_sum 加上当前草堆的重量进入下一轮循环
  • 本题解析所用的编程语言: Python
  • 完整且正确的编程代码
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 
# @param nums int整型一维数组 
# @return int整型
#
class Solution:
    def pivotIndex(self , nums: List[int]) -> int:
        n = len(nums)
        sum = 0
        for i in range(n):
            sum += nums[i]

        left_sum = 0
        for i in range(n):
            # 计算右边所有草堆的总重量
            right_sum = sum - nums[i] - left_sum
            if left_sum == right_sum:
                return i
            left_sum += nums[i]

        return -1
牛客高频top202题解系列 文章被收录于专栏

记录刷牛客高频202题的解法思路

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务