题解 | #相等的草堆#

相等的草堆

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

考察的知识点:数组;

解答方法分析:

  1. 遍历一次数组,求得整个数组的总重量totalWeight
  2. 遍历数组中的每一个元素,利用累加求和的思路计算左边草堆的总重量,并同时计算右边草堆的总重量。
  3. 对于每个草堆,如果左边草堆的总重量等于右边草堆的总重量,则找到了平衡点,返回当前草堆的下标。
  4. 如果遍历完所有草堆都没有找到平衡点,则返回-1。

所用编程语言:C++;

完整编程代码:↓

class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param nums int整型vector
     * @return int整型
     */
    int pivotIndex(vector<int>& grass) {
        int totalWeight = 0;
        int leftWeight = 0;

        for (int weight : grass) {
            totalWeight += weight;
        }

        for (int i = 0; i < grass.size(); i++) {
            if (leftWeight == totalWeight - leftWeight - grass[i]) {
                return i;
            }
            leftWeight += grass[i];
        }

        return -1;
    }
};

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务