构成特定和需要添加的最少元素
给你一个整数数组 nums ,和两个整数 limit 与 goal 。数组 nums 有一条重要属性:abs(nums[i]) <= limit 。
返回使数组元素总和等于 goal 所需要向数组中添加的 最少元素数量 ,添加元素 不应改变 数组中 abs(nums[i]) <= limit 这一属性。
注意,如果 x >= 0 ,那么 abs(x) 等于 x ;否则,等于 -x 。
示例 1:
输入:nums = [1,-1,1], limit = 3, goal = -4
输出:2
解释:可以将 -2 和 -3 添加到数组中,数组的元素总和变为 1 - 1 + 1 - 2 - 3 = -4 。
示例 2:
输入:nums = [1,-10,9,1], limit = 100, goal = 0
输出:1
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/minimum-elements-to-add-to-form-a-given-sum
代码区
class Solution:
def minElements(self, nums: List[int], limit: int, goal: int) -> int:
return ceil(abs(sum(nums)-goal)/limit)
解析
先判断数组nums的所有元素的和sum与目标goal的差距,然后除以limit,所得结果如果是整数,即为构成特定和需要添加的最少元素;如果是小数,所得结果需向上取整才得到想要结果。
其实不管是整数还是小数,向上取整操作后结果都是所求值,我们可以用ceil()函数来实现向上取整。
题目给出了abs()函数的功能:注意,如果 x >= 0 ,那么 abs(x) 等于 x ;否则,等于 -x 。
复杂度
时间复杂度:O(n),n为nums的长度,只遍历了一遍nums。
空间复杂度:O(1)