题目描述:
给你一个 非严格递增排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。
考虑 nums 的唯一元素的数量为 k ,你需要做以下事情确保你的题解可以被通过:
更改数组 nums ,使 nums 的前 k 个元素包含唯一元素,并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums 的大小不重要。
返回 k 。
判题标准:
系统会用下面的代码来测试你的题解:
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 。不需要考虑数组中超出新长度后面的元素。
思路一:
(1)设立一个flag变量用来与下一个元素比较,找到重复的元素,初始为nums[0]。
(2)从nums[1]开始遍历nums。
(3)将重复的元素索引存在一个索引列表中。
(4)遍历完成后,按照索引列表从后往前的顺序删除nums中的对应元素,此处必须是从后往前,否则索引会混乱。
代码如下:
class Solution(object):
def removeDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if len(nums) == 0:
return 0
else:
flag = nums[0]
idex_list = []
for i in range(1, len(nums)):
if nums[i] == flag:
flag = nums[i]
idex_list.append(i)
else:
flag = nums[i]
continue
for i in idex_list[::-1]:
nums.pop(i)
return len(nums)
执行用时如下,一般般:
思路二:来自官方思路——快慢指针:
此处直接展示官方的思路:
大概看了一下,感觉和自己的思路确实有区别,自己的思路除了遍历有序数组所用的时间之外,还耗费了时间遍历了重复数索引数组,以及数组的pop操作,花费时间更长。实现代码如下:
class Solution(object):
def removeDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if len(nums) == 0:
return 0
else:
i = 1 # i所指的这个数所在的前一个数是处理完成的数
for j in range(1, len(nums)):
# 如果处理完的数与遍历到的数不一样,说明新的数出现了,更新
if nums[i - 1] != nums[j]:
nums[i] = nums[j]
i += 1
else:
continue
return i
执行时间如下,快了很多: