题解 | #相等的草堆# 前缀和

相等的草堆

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

知识点

前缀和

思路

从左到右遍历,找到第一个左右两边的质量相等的位置,计算左右两边的质量可以计算前缀和优化每次计算到O(1)

时间复杂度为O(n)

AC code (C++)

#include <numeric>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int pivotIndex(vector<int>& nums) {
        int n = nums.size();
        int sum = accumulate(nums.begin(), nums.end(), 0);
        int last = 0;
        for (int i = 0; i < n; i ++) {
            int right = sum - nums[i] - last;
            if (last == right) return i;
            last += nums[i];
        }
        return -1;
    }
};

全部评论

相关推荐

07-02 22:46
门头沟学院 Java
码农索隆:hr:“管你投没投,先挂了再说”
点赞 评论 收藏
分享
点赞 评论 收藏
分享
07-22 11:07
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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