日渐头秃的代码日记 -- 6 LC第286周赛第二题

189 阅读2分钟

美化数组的最少删除数

给你一个下标从 0 开始的整数数组 nums ,如果满足下述条件,则认为数组 nums 是一个 美丽数组 :

  • nums.length 为偶数
  • 对所有满足 i % 2 == 0 的下标 i ,nums[i] != nums[i + 1] 均成立

注意,空数组同样认为是美丽数组。

你可以从 nums 中删除任意数量的元素。当你删除一个元素时,被删除元素右侧的所有元素将会向左移动一个单位以填补空缺,而左侧的元素将会保持 不变 。

返回使 nums 变为美丽数组所需删除的 最少 元素数目

 

示例 1:

输入: nums = [1,1,2,3,5]
输出: 1
解释: 可以删除 nums[0] 或 nums[1] ,这样得到的 nums = [1,2,3,5] 是一个美丽数组。可以证明,要想使 nums 变为美丽数组,至少需要删除 1 个元素。

示例 2:

输入: nums = [1,1,2,2,3,3]
输出: 2
解释: 可以删除 nums[0] 和 nums[5] ,这样得到的 nums = [1,2,2,3] 是一个美丽数组。可以证明,要想使 nums 变为美丽数组,至少需要删除 2 个元素。

 

提示:

  • 1 <= nums.length <= 105
  • 0 <= nums[i] <= 105
class Solution:
    def minDeletion(self, nums: List[int]) -> int:
        if len(nums) == 0:
            return 0
        length = len(nums)
        s = 0
        x = 0
        for i in range(length - 1):
            if i % 2 == x:
                if nums[i] == nums[i+1]:
                    x = 1-x
                    s = s + 1
        if (length - s) % 2 == 1:
            s = s + 1
        return s
                    

改题目有一个重要的信息条件是:最后数组偶数下标的数字不能与它后一个数字相同,但是是允许奇数下标与后一个相同的,同时要求最终结果是偶数,所以当遍历完一遍之后最后是奇数个,就把最后一个扔掉。 上述代码里最重要的是两行:

i % 2 == x
x = 1 - x

对奇偶数坐标的判断要区分开,根据删除了多少个元素而定,x要不停地翻转,但是本方法并不需要额外的数组空间,只需要保存一个x和最终的结果即可。时间复杂度和空间复杂度都尚可。