开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第34天,点击查看活动详情
力扣——1785. 构成特定和需要添加的最少元素
1785. 构成特定和需要添加的最少元素
给你一个整数数组 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 <= 1051 <= limit <= 106-limit <= nums[i] <= limit-109 <= goal <= 109
问题解析
我们先计算出数组的总和sum,再看sum和目标值goal的差值sub是多少。
为了方便处理,我们可以不用管差值sub的正负性,因为我们加入的新的元素的大小是看它的绝对值是否小于等于limit的,所以正负性没有必要,为了方便我们直接计算差值sub的绝对值。
然后为了让加入的新元素的个数尽量小,我们可以先加绝对值为limit的数,当差值sub不够limit后,再加一个数进去就可以使得数组和等于goal了。但是这一题的数很大,我们一个个的模拟加太慢了会超时,我们直接通过除法计算需要加多少个数即可。
还要注意的是,这一题数很大,最大的数组和达到了1e9*1e6,远远超过了int的存储范围,所以我们在计算过程中要开longlong,不然会爆数据。
AC代码
class Solution {
public:
int minElements(vector<int>& nums, int limit, int goal) {
long long sum=0;
for(auto&i:nums)
{
sum+=i;
}
long long sub=abs(sum-goal);
long long cnt=sub/limit;
if(sub%limit!=0)cnt++;
return cnt;
}
};