刪除排序數組中的重復項


給定一個排序數組,你需要在原地刪除重復出現的元素,使得每個元素只出現一次,返回移除后數組的新長度。

不要使用額外的數組空間,你必須在原地修改輸入數組並在使用 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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM