LeeCode-使数组中所有元素都等于零(python)
给你一个非负整数数组 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
# -*- coding: utf-8 -*-
# @Time : 2022/7/6 14:51
# @Author : 凌贤文
# @Email : lingxw@zjnu.edu.cn
from typing import List
class Solution:
def minimumOperations(self, nums: List[int]) -> int:
# 返回计数参数count
count = 0
while True:
# 复制数组为x
x = nums.copy()
nums = set(x)
# 移除数组中的0
if 0 in nums:
nums.remove(0)
#
if len(nums) == 0:
break
n = min(nums)
# 数组减去每次最小的;
# 值得学习的步骤;
nums = [i - n for i in nums]
count += 1
return count
if __name__ == __main__:
sol = Solution()
print(sol.minimumOperations([1, 5, 0, 3, 5]))
