题目描述
给你一个整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-elements-to-add-to-form-a-given-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
class Solution {
public:
int minElements(vector<int>& nums, int limit, int goal) {
int times = 0;
long long cursum = 0;
for(auto &it : nums) cursum += it;
long long lgoal = goal - cursum;
while(lgoal){
long long add = min((long long)limit, abs(lgoal));
int tmp = abs(lgoal) / add;
if(goal > 0) lgoal -= add * tmp;
else lgoal += add * tmp;
times += tmp;
}
return times;
}
};