https://leetcode.cn/problems/find-pivot-index/
题目要求
给你一个整数数组 nums ,请计算数组的 中心下标 。
数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果中心下标位于数组最左端,那么左侧数之和视为 0 ,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1 。
遍历
class Solution {
public int pivotIndex(int[] nums) {
int n = nums.length;
int sum = 0;
for (int num : nums) {
sum += num;
}
int l = 0, r = sum - nums[0];
if (l == r) return 0;
for (int i = 1; i < n; i++) {
l += nums[i - 1];
r -= nums[i];
if (l == r) return i;
}
return -1;
}
}
- 遍历求出总和sum
- 再遍历求中心索引左半和leftSum
- 同时根据sum和 L 计算中心索引右半和 R
- 判断 L 和 R 是否相同