坏蛋Dan
知乎@坏蛋Dan
发布时间:2024.2.4

前言

今天有两道题,其中一道是另一道的变体,所以放到一起


题目1

给你一个 非严格递增排列 的数组 nums ,请你** 原地** 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。

考虑 nums 的唯一元素的数量为 k ,你需要做以下事情确保你的题解可以被通过:

  • 更改数组 nums ,使 nums 的前 k 个元素包含唯一元素,并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums 的大小不重要。
  • 返回 k

判题标准:

系统会用下面的代码来测试你的题解:

如果所有断言都通过,那么您的题解将被 通过

示例 1:

示例 2:

提示:

  • 1 <= nums.length <= 3 * 104
  • -104 <= nums[i] <= 104
  • nums 已按 非严格递增 排列

分析

这里拿到的数组已经是递增排序了的,所以我们可以直接nums[i] == nums[i - 1]来判断是否是相等的,如果是,那就是重复的元素,直接移除即可。


没啥好说的,另外这里应该不用移除元素,题目只需要我们将新的长度返回即可,所以可以使用最初的len - count


测试用例

然后我们来看下变体


题目2

在上面的基础下:

给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。

不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。


分析

需要我们使用O(1)的空间复杂度,也就是不能使用一维以上的结构。

这个简单,我们再搞个变量用来表示重复次数即可。


没啥好说的


测试用例


总结

这题没啥好说的,主要是给的数组已经是排序了的,如果没有排序,那么我们的空间复杂度要到O(n),因为相同元素不连续,所以我们还得记录元素出现的个数。