0 offer : 腾讯云智也是腾讯,运营开发也是开发

2024-05-13 16:42:56 浏览数 (3)

回归主线。

五一假期,给大家安排一些「难度不大」的算法题。

题目描述

平台:LeetCode

题号:1785

给你一个整数数组 nums,和两个整数 limitgoal

数组 nums 有一条重要属性:abs(nums[i]) <= limit

返回使数组元素总和等于 goal 所需要向数组中添加的 最少元素数量 ,添加元素 不应改变 数组中 abs(nums[i]) <= limit 这一属性。

注意,如果 x >= 0,那么 abs(x) 等于 x ;否则,等于 -x

示例 1:

代码语言:javascript复制
输入:nums = [1,-1,1], limit = 3, goal = -4

输出:2

解释:可以将 -2 和 -3 添加到数组中,数组的元素总和变为 1 - 1   1 - 2 - 3 = -4 。

示例 2:

代码语言:javascript复制
输入:nums = [1,-10,9,1], limit = 100, goal = 0

输出:1

提示:

1 <= nums.length <= 10^5
1 <= limit <= 10^6
-limit <= nums[i] <= limit
-10^9 <= goal <= 10^9

贪心

对于 nums 而言,我们可以先通过

O(n)

的遍历求得其原总和 sum 为何值。

sumgoal 不等,我们可以按照「贪心」的方式里添加元素。

由于添加的元素需要满足 abs(x) <= limit 要求,因此我们添加数的范围在

[-limit, limit]

之间。

为确保添加的元素最小,我们应当优先添加能够有效抵消两者差值的数值(添加

limit

-limit

),添加个数为

left lceil frac{left | sum - goal right |}{limit} right rceil

Java 代码:

代码语言:javascript复制
class Solution {
    public int minElements(int[] nums, int limit, int goal) {
        long sum = 0;
        for (int x : nums) sum  = x;
        return (int) ((Math.abs(sum - goal)   limit - 1) / limit);
    }
}

TypeScript 代码:

代码语言:javascript复制
function minElements(nums: number[], limit: number, goal: number): number {
    let sum = 0
    for (const x of nums) sum  = x
    return Math.ceil(Math.abs(sum - goal) / limit)
}

Python 代码:

代码语言:javascript复制
class Solution:
    def minElements(self, nums: List[int], limit: int, goal: int) -> int:
        sumv = sum(nums)
        return (abs(sumv - goal)   limit - 1) // limit
  • 时间复杂度:
O(n)
  • 空间复杂度:
O(1)

0 人点赞