昨天懈怠了一天,今天赶紧把昨天的每日一题补上!
题目描述
给你一个整数数组 nums 。你可以选定任意的 正数 startValue 作为初始值。
你需要从左到右遍历 nums 数组,并将 startValue 依次累加上 nums 数组中的值。
请你在确保累加和始终大于等于 1 的前提下,选出一个最小的 正数 作为 startValue 。
示例:
输入:nums = [-3,2,-3,4,2]
输出:5
解释:如果你选择 startValue = 4,在第三次累加时,和小于 1 。
累加求和
startValue = 4 | startValue = 5 | nums
(4 -3 ) = 1 | (5 -3 ) = 2 | -3
(1 +2 ) = 3 | (2 +2 ) = 4 | 2
(3 -3 ) = 0 | (4 -3 ) = 1 | -3
(0 +4 ) = 4 | (1 +4 ) = 5 | 4
(4 +2 ) = 6 | (5 +2 ) = 7 | 2
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/minimum-value-to-get-positive-step-by-step-sum
2.思路
- 从暴力方法入手,从最小的正整数1开始,依次和数组中的每一项相加得到结果,找到满足条件的最小正数即可,但这样做时间复杂度很高;
- 考虑到使用前缀和:1)当前缀和>=1的时候,取最小的正数即可;2)当前缀和<0的时候,找出绝对值最大的前缀和 + 1即可,需要使用一个flag标记是否出现了前缀和<0的情况;
3.代码
int minStartValue(vector<int>& nums) {
/*
- 首先想到的方法是遍历数组中的每一项,判断其是否满足条件,并且在满足条件中的所有项目中找到最小的正数即可;
- 但是忽略了当数组中全是负数的情况,因此该方案是不可行的;
- 累加和 >= 1,当数组中出现负数的时候会影响值;
- 前缀和,当前缀和小于0的时候,取前缀和的绝对值,将所有这样的数字相比较找到最小的正数 + 1 就是答案
- 若前缀和 >= 1,则取最小的前缀和即可
*/
if(nums.size() == 1){
return nums[0] <= 0 ? abs(nums[0]+1) : nums[0];
}
int pre = 0;
int res = 1; // 设置初始值为最小的正整数
bool flag = 0;
for(int i = 0; i < nums.size();i++){
pre += nums[i];
if(pre < 0){
res = max(res, abs(pre) + 1);
flag = 1;
}
else{
if(flag == 0){
if(pre == 0){
res = 1;
}
else{
res = min(res, pre);
}
}
}
}
return res;
}