给你一个整数数组 nums
,请计算数组的 中心下标 。
数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果中心下标位于数组最左端,那么左侧数之和视为 0
,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1
示例 1:
输入:nums = [1, 7, 3, 6, 5, 6] 输出:3 解释: 中心下标是 3 。 左侧数之和 sum = nums[0] + nums[1] + nums[2] = 1 + 7 + 3 = 11 , 右侧数之和 sum = nums[4] + nums[5] = 5 + 6 = 11 ,二者相等。
示例 2:
输入:nums = [1, 2, 3] 输出:-1 解释: 数组中不存在满足此条件的中心下标。
示例 3:
输入:nums = [2, 1, -1] 输出:0 解释: 中心下标是 0 。 左侧数之和 sum = 0 ,(下标 0 左侧不存在元素), 右侧数之和 sum = nums[1] + nums[2] = 1 + -1 = 0 。
提示:
1 <= nums.length <= 10^4
-1000 <= nums[i] <= 1000
注意:本题与 LeetCode 1991 题相同:LeetCode 1991. 找到数组的中间位置-CSDN博客
提示 1
Create an array sumLeft where sumLeft[i] is the sum of all the numbers to the left of index i.
提示 2
Create an array sumRight where sumRight[i] is the sum of all the numbers to the right of index i.
提示 3
For each index i, check if sumLeft[i] equals sumRight[i] return i. If no i found, return -1.
解法:一维前缀和
这题超简单,提示讲的很明白。🎉
Java版:
class Solution {
public int pivotIndex(int[] nums) {
int n = nums.length;
int[] prefix = new int[n + 1];
for (int i = 1; i <= n; i++) {
prefix[i] = prefix[i - 1] + nums[i - 1];
}
for (int i = 0; i < n; i++) {
// nums[0..i..n - 1]
// sumLeft[i]: prefix[i], sumRight[i]: prefix[n] - prefix[i + 1]
if (prefix[i] == prefix[n] - prefix[i + 1]) {
return i;
}
}
return -1;
}
}
复杂度分析
- 时间复杂度:O(n),n 是 数组 nums 的长度。总的时间复杂度 O(2n) = O(n)。
- 空间复杂度:O(n),n 是 数组 nums 的长度。
Python3版:
class Solution:
def pivotIndex(self, nums: List[int]) -> int:
prefix = []
prefix.append(0)
n = len(nums)
for i in range(1, n + 1):
prefix.append(prefix[i - 1] + nums[i - 1])
for i in range(0, n):
if prefix[i] == prefix[n] - prefix[i + 1]:
return i
return -1
复杂度分析
- 时间复杂度:O(n),n 是 数组 nums 的长度。总的时间复杂度 O(2n) = O(n)。
- 空间复杂度:O(n),n 是 数组 nums 的长度。
同类题型:
LeetCode 1991. 找到数组的中间位置-CSDN博客
LeetCode 2574. 左右元素和的差值-CSDN博客