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

最新推荐文章于 2024-07-14 02:25:25 发布

Schanappi

最新推荐文章于 2024-07-14 02:25:25 发布

阅读量59

点赞数

分类专栏： LeetCode刷题文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_43894455/article/details/128338738

版权

LeetCode刷题专栏收录该内容

218 篇文章 6 订阅

订阅专栏

题目描述

给你一个整数数组 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 <= 10⁵
1 <= limit <= 10⁶
-limit <= nums[i] <= limit
-10⁹ <= goal <= 10⁹

方法一：模拟，向上取整

思路：

首先计算数组元素的总和 sum ，将 sum 和 goal 相减就得到二者差距「diff = abs( sum - goal ) 」，因此题目就转换成为如何操作使得 diff = 0 ，为了得到最小操作次数，让差距除以 limit 并向上取整即可。
如何实现向上取整呢？
如果 diff % limit == 0（能整除），此时 cnt = diff / limit ；
如果 diff % limit !=0（不能整除），此时 cnt = diff / limit + 1 ;
综合来看，cnt = （ diff + limit -1 ）/ limit ；

情况

通过；

收获

我还是把这道题想麻烦了，居然还考虑到动态规划，其实远比我想象的来得简单；
看到提示， goal 的取值范围很大，因此 sum 必须使用 long long ，否则无法通过；
通过这道题记住了如何实现向上取整。

时间复杂度：O（n），n 为数组长度；
空间复杂度：O（1）

class Solution {
public:
    int minElements(vector<int>& nums, int limit, int goal) {
        long long sum = 0;
        int cnt = 0;
        for(auto& num : nums)   sum += num;
        cnt = (abs(sum - goal) + limit - 1) / limit;
        return cnt;
    }
};