題目描述:
給定一個排序數組,你需要在原地刪除重復出現的元素,使得每個元素只出現一次,返回移除后數組的新長度。
不要使用額外的數組空間,你必須在原地修改輸入數組並在使用 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
。
你不需要考慮數組中超出新長度后面的元素。
題目分析:
看到這題目,不看題目的具體要求,第一想到的是用set函數,set和dict類似,也是一組key的集合,但不存儲value。由於key不能重復,所以,在set中,沒有重復的key。利用這一特性,可以過濾掉重復元素。
但是仔細一看題目要求,完全不符合,題目是要求原地刪除重復元素。如果用set方法,會返回一個新的字典,而原數組沒有改變,結果如下:
>>> nums = [1,1, 2, 3, 3, 3, 5, 6]
>>> s = set(nums)
>>> nums
[1, 1, 2, 3, 3, 3, 5, 6]
>>> s
{1, 2, 3, 5, 6}
跑偏了,回到題目本身。
list有個remove方法,remove方法是一個沒有返回值的原位置改變的方法,他修改了列表但是沒有返回值。
由於給定的是排序數組,這就容易多了,比較兩兩相鄰的數組元素,相同就把其中的一個利用remove方法從列表中移除。
class Solution(object): def removeDuplicates(self, nums): """ :type nums: List[int] :rtype: int """ list_index = 0 while list_index < len(nums) - 1: if nums[list_index] == nums[list_index + 1]: nums.remove(nums[list_index]) else: list_index += 1
return len(nums)