力扣——1785. 构成特定和需要添加的最少元素

101 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第34天,点击查看活动详情

力扣——1785. 构成特定和需要添加的最少元素

1785. 构成特定和需要添加的最少元素

给你一个整数数组 nums ,和两个整数 limitgoal 。数组 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

问题解析

我们先计算出数组的总和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;
    }
};