-
Notifications
You must be signed in to change notification settings - Fork 4
/
remove_duplicates_from_sorted_array.py
83 lines (60 loc) · 2.52 KB
/
remove_duplicates_from_sorted_array.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
"""
26. 删除有序数组中的重复项
数组 双指针
简单
给你一个 升序排列 的数组 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 * 10^4
-10^4 <= nums[i] <= 10^4
nums 已按 升序 排列
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
"""
from typing import List
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
pre = -10001
unique_list = []
for i in range(len(nums)):
n = nums[i]
if n != pre:
unique_list.append(n)
pre = n
for j in range(len(unique_list)):
nums[j] = unique_list[j]
return len(unique_list)
if __name__ == '__main__':
solution = Solution()
num_list = [1, 1, 2]
result = solution.removeDuplicates(num_list)
print(result, num_list)
assert result == 2
assert num_list[:result] == [1, 2]
num_list = [0, 0, 1, 1, 1, 2, 2, 3, 3, 4]
result = solution.removeDuplicates(num_list)
print(result, num_list)
assert result == 5
assert num_list[:result] == [0, 1, 2, 3, 4]