刷题的日常-逐步求和得到正数的最小值

79 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第14天,点击查看活动详情

刷题的日常

一天一题,保持脑子清爽

逐步求和得到正数的最小值

来自leetcode的 1413 题,题意如下:

给你一个整数数组 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

理解题意

我们可以从题意中提取的条件如下:

  • 题目给定一个数组,数组内的元素是无序的
  • 要求我们从左到右依次将所有的数加起来,这时有个限制规则
    • 要求我们找一个数
    • 这个数和已经扫描过的元素相加
    • 相加的总和始终大于等于1
  • 如果数组内的总和始终大于等于1,要求返回的数必须是最小的正整数1

做题思路

其实很简单,就是在扫描的过程中,逐步汇总数组的和,这里可以使用前缀和来做

  • 使用sum来汇总数组的前缀和
  • 用result来代表返回的最小正整数
  • 在遍历的过程中,如果出现了总和小于0的情况,就直接取小于0的部分
  • 比较小于0的部分和之前的最小值做比较,取最大的一个
  • 如果最终结果等于0,则返回1,否则返回结果

代码实现

代码实现如下:

public class Solution {
    public int minStartValue(int... nums) {
        int sum = -1;
        int result = 0;
        for (int num : nums) {
            sum += num;
            if (sum < 0) {
                result = Math.max(-sum, result);
            }
        }
        return result == 0 ? 1 : result;
    }
}