給定一個排序數組,你需要在原地刪除重復出現的元素,使得每個元素只出現一次,返回移除后數組的新長度。
不要使用額外的數組空間,你必須在原地修改輸入數組並在使用 O(1) 額外空間的條件下完成。
示例 1:
給定數組 nums = [1,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
。 你不需要考慮數組中超出新長度后面的元素。
說明:
為什么返回數值是整數,但輸出的答案是數組呢?
請注意,輸入數組是以“引用”方式傳遞的,這意味着在函數里修改輸入數組對於調用者是可見的。
你可以想象內部操作如下:
// nums 是以“引用”方式傳遞的。也就是說,不對實參做任何拷貝 int len = removeDuplicates(nums); // 在函數里修改輸入數組對於調用者是可見的。 // 根據你的函數返回的長度, 它會打印出數組中該長度范圍內的所有元素。 for (int i = 0; i < len; i++) { print(nums[i]); }
思路:
(1)重點是排序數組,如果這道題沒有排序的話,時間復雜度就應該是O(nlogn)級別的,由於排序了,就可以做到O(n)級別;
(2)另外由於要求在在使用 O(1) 額外空間的條件下完成,所以就不用考慮用字典來記錄哪些元素出現了重復;由於是排序數組,所以如果出現重復,應該也是前后位置重復;為了節約時間,不要用刪除列表元素的方式來清除重復元素,而是采用覆蓋元素的方法確保前n位沒有重復即可,所以可以采用如下代碼完成。
1 class Solution(object): 2 def removeDuplicates(self, nums): 3 """ 4 :type nums: List[int] 5 :rtype: int 6 """ 7 if len(nums) == 0: 8 return 0 9 location = 1 10 for index,item in enumerate(nums): 11 if index == 0: 12 continue 13 if item != nums[index-1]: 14 nums[location] = item 15 location = location + 1 16 return location