LeetCode——2357. 使数组中所有元素都等于零

一、题目

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

选出一个正整数 x ，x 需要小于或等于 nums 中 最小 的 非零 元素。
nums 中的每个正整数都减去 x。
返回使 nums 中所有元素都等于 0 需要的 最少 操作数。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/make-array-zero-by-subtracting-equal-amounts/description/

二、C++解法

我的思路及代码

排序+模拟

对数组进行排序后，每次让非零元素减去最小的非零元素，直到数组全部都为 0 为止。

class Solution {
public:
    int minimumOperations(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        int count = 0;
        int index = 0;
        while(index<nums.size()&&nums[index]==0)
            index++;
        if(index == nums.size()) return 0;
        while(nums[nums.size()-1]!=0){
            int temp = nums[index];
            if(temp!=0){
                for(int i=index;i<nums.size();i++){
                    nums[i] -= temp;
                    if(nums[i] == 0)
                        index++;
                }
            }
            count++;
        }
        return count;
    }
};

• 时间复杂度：O(n²)，其中 nnn 是数组 nums 的长度。排序需要 O(nlog⁡n) 的时间，排序之后需要遍历数组一次，对于每个非零元素，将数组中的所有非零元素减去最小非零元素需要 O(n) 的时间，因此时间复杂度是 O(n²)
• 空间复杂度：O(log⁡n)，其中 nnn 是数组 nums 的长度。排序需要 O(log⁡n) 的递归调用栈空间

官方参考代码

哈希集合

class Solution {
public:
    int minimumOperations(vector<int>& nums) {
        unordered_set<int> hashSet;
        for (int num : nums) {
            if (num > 0) {
                hashSet.emplace(num);
            }
        }
        return hashSet.size();
    }
};

• 时间复杂度：O(n)，其中 nnn 是数组 nums 的长度。需要遍历数组一次，每个非零元素加入哈希集合的时间是 O(1)
• 空间复杂度：空间复杂度：O(n)，其中 n 是数组 nums 的长度。哈希集合需要 O(n) 的空间