使数组中所有元素都等于零

301 阅读1分钟

这是我参与「掘金日新计划 · 2 月更文挑战」的第 26 天,点击查看活动详情

问题描述

给你一个非负整数数组 nums 。在一步操作中,你必须:

  • 选出一个正整数 x ,x 需要小于或等于 nums 中 最小 的 非零 元素。
  • nums 中的每个正整数都减去 x

返回使 nums 中所有元素都等于 **0 需要的 最少 操作数。

示例 1:

输入: nums = [1,5,0,3,5]
输出: 3
解释:
第一步操作:选出 x = 1 ,之后 nums = [0,4,0,2,4] 。
第二步操作:选出 x = 2 ,之后 nums = [0,2,0,0,2] 。
第三步操作:选出 x = 2 ,之后 nums = [0,0,0,0,0] 。

示例 2:

输入: nums = [0]
输出: 0
解释: nums 中的每个元素都已经是 0 ,所以不需要执行任何操作。

提示:

  • 1 <= nums.length <= 100
  • 0 <= nums[i] <= 100

思路分析

首先我们先要理解一下题目意思,题目会给我们一个数组,我们需要把数组中的所有元素值变为0,我们每次可以对数组进行以下操作:

  • 1、选出一个正整数 x ,x 需要小于或等于 nums 中 最小 的 非零 元素。
  • 2、nums 中的每个正整数都减去 x

简单来说,就是每次我们可以将数组中非零的元素减去数组中非零的最小值,所以我们每次操作需要先找出数组中非零的最小值,然后将数组中所有值大于0的元素值减去这个最小值,不断循环这一步操作,知道数组中所有值都变为0,我们需要计算将数组所有元素都变为零所需进行的步数。

  • 找出数组中非零的最小值

我们可以直接使用Math.min来快速找到数组中的最小值,因为我们这里是需要找到不为零的最小值,所以我们可以先将零置为最大值。

for(let i = 0; i < nums.length; i++){
    if(nums[i] == 0) nums[i] = Infinity;
}
let min = Math.min(...nums);
  • 对数组元素进行减去最小值操作

这里我们将零都置为了Infinity,所以我们应该在数组中最小值为Infinity时结束循环。

let res = 0;
while(min != Infinity){
    res++;
    for(let i = 0; i < nums.length; i++){
        if(nums[i] > 0) nums[i] -= min;
        if(nums[i] == 0) nums[i] = Infinity;
    }
    min = Math.min(...nums);
}

完整 AC 代码如下:

AC 代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var minimumOperations = function(nums) {
    for(let i = 0; i < nums.length; i++){
        if(nums[i] == 0) nums[i] = Infinity;
    }
    let min = Math.min(...nums);
    let res = 0;
    while(min != Infinity){
        res++;
        for(let i = 0; i < nums.length; i++){
            if(nums[i] > 0) nums[i] -= min;
            if(nums[i] == 0) nums[i] = Infinity;
        }
        min = Math.min(...nums);
    }
    return res;
};

说在后面

本人为算法业余爱好者,平时只是随着兴趣偶尔刷刷题,如果上面分享有错误的地方,欢迎指出,感激不尽。