给你一个下标从 0 开始的整数数组 nums
,请你找到 最左边 的中间位置 middleIndex
(也就是所有可能中间位置下标最小的一个)。
中间位置 middleIndex
是满足 nums[0] + nums[1] + ... + nums[middleIndex-1] == nums[middleIndex+1] + nums[middleIndex+2] + ... + nums[nums.length-1]
的数组下标。
如果 middleIndex == 0
,左边部分的和定义为 0
。类似的,如果 middleIndex == nums.length - 1
,右边部分的和定义为 0
。
请你返回满足上述条件 最左边 的 middleIndex
,如果不存在这样的中间位置,请你返回 -1
。
示例 1:
输入:nums = [2,3,-1,8,4] 输出:3 解释: 下标 3 之前的数字和为:2 + 3 + -1 = 4 下标 3 之后的数字和为:4 = 4
示例 2:
输入:nums = [1,-1,4] 输出:2 解释: 下标 2 之前的数字和为:1 + -1 = 0 下标 2 之后的数字和为:0
示例 3:
输入:nums = [2,5] 输出:-1 解释: 不存在符合要求的 middleIndex 。
示例 4:
输入:nums = [1] 输出:0 解释: 下标 0 之前的数字和为:0 下标 0 之后的数字和为:0
提示:
1 <= nums.length <= 100
-1000 <= nums[i] <= 1000
注意:本题与 LeetCode 724 题相同:LeetCode 724. 寻找数组的中心下标-CSDN博客
提示 1
Could we go from left to right and check to see if an index is a middle index?
提示 2
Do we need to sum every number to the left and right of an index each time?
提示 3
Use a prefix sum array where prefix[i] = nums[0] + nums[1] + ... + nums[i].
解法:一维前缀和
这题与 724 题相同,但是724题的题目讲的更明白,题意更明显。
class Solution {
public int findMiddleIndex(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 的长度。
同类题型:
LeetCode 724. 寻找数组的中心下标-CSDN博客
LeetCode 2574. 左右元素和的差值-CSDN博客