题解 | #相等的草堆#
相等的草堆
https://www.nowcoder.com/practice/0e2f3b27bbdc45fcbc70cc4fd41e15fe
考察的知识点:数组;
解答方法分析:
- 遍历一次数组,求得整个数组的总重量
totalWeight
。 - 遍历数组中的每一个元素,利用累加求和的思路计算左边草堆的总重量,并同时计算右边草堆的总重量。
- 对于每个草堆,如果左边草堆的总重量等于右边草堆的总重量,则找到了平衡点,返回当前草堆的下标。
- 如果遍历完所有草堆都没有找到平衡点,则返回-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; } };