世界上并没有完美的程序,但是我们并不因此而沮丧,因为写程序就是一个不断追求完美的过程。
-侯氏工坊
题目
给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么nums的前 k 个元素应该保存最终结果。
将最终结果插入nums 的前 k 个位置后返回 k 。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
判题标准:
系统会用下面的代码来测试你的题解:
int[] nums = [...]; // 输入数组
int[] expectedNums = [...]; // 长度正确的期望答案
int k = removeDuplicates(nums); // 调用
assert k == expectedNums.length;
for (int i = 0; i < k; i++) {
assert nums[i] == expectedNums[i];
}
如果所有断言都通过,那么您的题解将被 通过。
示例 1:
输入:nums = [1,1,2]
输出:2, nums = [1,2,_]
解释:函数应该返回新的长度 2 ,并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出新长度后面的元素。
示例 2:
输入:nums = [0,0,1,1,1,2,2,3,3,4]
输出:5, nums = [0,1,2,3,4]
解释:函数应该返回新的长度 5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
提示:
0 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按 升序 排列
思路
- 如果nums的长度为0或1,直接返回其长度即可
- 记录nums的起始值pre=nums[0],记录一个起始索引i=0
- 因为是升序,那么后面的值要么大于pre,要么等于pre
- 等于pre,说明是重复值,不予理会
- 大于pre,则将当前值赋值给pre,并且索引i+1,将当前值赋值给nums[i]
- 这样nums前面的值都是递增的而没有重复项
- 非重复的长度为最终的索引值加1,即i+1
实现
def remove_duplicates(nums):
nums_len = len(nums)
if nums_len in range(2):
return nums_len
i = 0
pre = nums[0]
for num in nums:
if pre < num:
pre = num
i += 1
nums[i] = pre
return i + 1
- 测试
assert 4 == remove_duplicates([1, 2, 3, 3, 3, 4])
- 测试通过