寫在前面：看到很多程序員朋友都在刷LeetCode,剛好最近在學習swift以及數(shù)據(jù)結(jié)構(gòu)算法，所以用swift刷LeetCode應該是個很好的練習方式。自己會在playground上編寫代碼及測試，并上傳github持續(xù)更新，地址：LeetCode-Swift

題目描述

給定一個排序數(shù)組，你需要在原地刪除重復出現(xiàn)的元素，使得每個元素只出現(xiàn)一次，返回移除后數(shù)組的新長度。
不要使用額外的數(shù)組空間，你必須在原地修改輸入數(shù)組并在使用 O(1) 額外空間的條件下完成。

示例1

給定數(shù)組 nums = [1,1,2], 

函數(shù)應該返回新的長度 2, 并且原數(shù)組 nums 的前兩個元素被修改為 1, 2。 

你不需要考慮數(shù)組中超出新長度后面的元素。

示例2

給定 nums = [0,0,1,1,1,2,2,3,3,4],

函數(shù)應該返回新的長度 5, 并且原數(shù)組 nums 的前五個元素被修改為 0, 1, 2, 3, 4。

你不需要考慮數(shù)組中超出新長度后面的元素。

審題

• 有序可變數(shù)組，返回刪除重復元素后的數(shù)組的長度
• 不使用額外的數(shù)組空間，即不能創(chuàng)建新的數(shù)組變量
• 最后可變數(shù)組變?yōu)橐粋€不含重復元素的有序數(shù)組

代碼

func removeDuplicates(_ nums: inout [Int]) -> Int {
    
    if nums.count <= 1 {
        return nums.count
    }
    
    var i = 0
    for j in 0..<nums.count {
        if nums[j] != nums[i] {
            i += 1
            nums[i] = nums[j]
        }
    }
    
    if i != nums.count {
        for _ in (i+1) ..< nums.count {
            nums.remove(at: i+1)
        }
    }
    
    return i + 1;
}

測試

var nums0 = [0]
removeDuplicates(&nums0)   // 1
print(nums0)               // [0] 

var nums1 = [0,1,1,2,2,2,3,3,4]
removeDuplicates(&nums1)  // 5
print(nums1)              //[0,1,2,3,4]

說明

1、題目要求輸入?yún)?shù)可以被修改，所以用了inout關(guān)鍵字（輸入輸出參數(shù)）
2、nums為指針變量，函數(shù)調(diào)用時記得前面加上&（手寫別忘了）
3、題目要求數(shù)組要原地刪除重復的元素，也就是如果存在重復元素，函數(shù)調(diào)用結(jié)束后輸入數(shù)組應該有變化，所以應該有下面這段代碼：

 if i != nums.count {
        for _ in (i+1) ..< nums.count {
            nums.remove(at: i+1)
        }
    }

網(wǎng)上一些解答都沒有這步處理，我覺得是不對的。

好了，這道題就到這，有什么疑問歡迎交流討論~