给你一个整数数组 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
提示:
1 <= nums.length <= 105
1 <= limit <= 106
-limit <= nums[i] <= limit
-109 <= goal <= 109
解法一:模拟
对数组求和后求出与goalgoalgoal的差距target=abs(sum−goal)target = abs(sum-goal)target=abs(sum−goal),我们的目标是使target变为0。
对于添加的数由于必须满足abs(nums[i])<=limitabs(nums[i])<=limitabs(nums[i])<=limit,因此能够放置最大的数便放置最大的数,次数为target/limittarget / limittarget/limit,若不能整除则还要多一次。
- 时间复杂度:O(n)O(n)O(n)
- 空间复杂度:O(1)O(1)O(1)
class Solution {
public int minElements(int[] nums, int limit, int goal) {
long sum = 0;
for (int x : nums) sum += x;
sum = Math.abs(goal - sum) + limit - 1;
return (int)(sum / limit);
}
}
class Solution {
public:
int minElements(vector<int>& nums, int limit, int goal) {
long long sum = goal;
for (int x : nums) sum -= x;
return (abs(sum) + limit - 1) / limit;
}
};