给你一个整数数组 nums
。你可以选定任意的 正数 startValue 作为初始值。
你需要从左到右遍历 nums
数组,并将 startValue 依次累加上 nums
数组中的值。
请你在确保累加和始终大于等于 1 的前提下,选出一个最小的 正数 作为 startValue 。
这道题可以用前缀和 + 贪心来解决
思路如下
咱们假设最小值为x,有一个数组 nums = [-3,2,-3,4,2]
那么应该有
x - 3 >= 1
x - 3 + 2 >= 1
...
x - 3 + 2 - 3 + 4 + 2 >= 1
我们用一个数组 a 来存储前缀和 a[i] 表示第 i 个下标的前缀和
则有 a = {-3, -1, -4, 0, 2}
此时,我们需要求出的是前缀和中的最小值,根据小学学习的数学公式
startValue + a.min >= 1
得出 startValue >= 1 - a.min
易有 最小值 startValue = 1 - a.min
根据上述条件,得出以下代码
public int minStartValue(int[] nums) {
int n = nums.length;
int[] a= new int[n]; // 前缀和数组
a[0] = nums[0];
for (int i = 1; i < n; i++) {
a[i] = a[i - 1] + nums[i];
}
int min = Integer.MAX_VALUE;
for (int i : a) { // 求出最小值
min = Math.min(i, min);
}
return min > 0 ? 1 : 1 - min; // 如果最小值大于0 直接返回1
}
在本题中,还存在优化的空间,再求前缀和最小值时,可在第一个循环中判断
代码如下
public int minStartValue(int[] nums) {
int n = nums.length;
int[] a= new int[n];
a[0] = nums[0];
int min = nums[0];
for (int i = 1; i < n; i++) {
a[i] = a[i - 1] + nums[i];
min = Math.min(min, a[i]);
}
return min > 0 ? 1 : 1 - min;
}
然后我们发现我们用到的只是当前下标的前缀和,那么我们可以用一个变量来存储当前下标前缀和,做到空间复杂度的优化
代码如下
public int minStartValue(int[] nums) {
int n = nums.length;
int prev = nums[0];
int min = nums[0];
for (int i = 1; i < n; i++) {
prev = prev + nums[i];
min = Math.min(min, prev);
}
return min > 0 ? 1 : 1 - min;
}