算法（1）- 刪除有序數組中的重復項

題目

給你一個有序數組 nums ，請你 原地 刪除重復出現的元素，使每個元素 只出現一次 ，返回刪除后數組的新長度。

不要使用額外的數組空間，你必須在 原地 修改輸入數組 並在使用 O(1) 額外空間的條件下完成

 

說明

為什么返回數值是整數，但輸出的答案是數組呢?

請注意，輸入數組是以「引用」方式傳遞的，這意味着在函數里修改輸入數組對於調用者是可見的。

你可以想象內部操作如下:

// nums 是以“引用”方式傳遞的。也就是說，不對實參做任何拷貝
int len = removeDuplicates(nums);

// 在函數里修改輸入數組對於調用者是可見的。
// 根據你的函數返回的長度, 它會打印出數組中 該長度范圍內 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[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 * 104
-104 <= nums[i] <= 104
nums 已按升序排列

 

正確答案

def test(nums: List[int])->int:
    # 左指針 i
    i = 0
    lens = len(nums)
    # 右指針 j
    for j in range(1,lens):
        # 如果不相等，則將左指針后移一位，並且賦值右指針的值
        if nums[i] != nums[j]:
            i +=1
            nums[i]=nums[j]
        # 如果左右指針的值相同，證明有重復的值，不做任何操作，右指針后移一位，左指針不動
        
    # 最終數組的長度就是左指針的位置+1
    return i+1

 

解題思路：雙指針

• 首先注意數組是有序的，那么重復的元素一定會相鄰
• 要求刪除重復元素，實際上就是將不重復的元素移到數組的左側

 

使用雙指針思路，左指針記作 i 也叫慢指針，右指針記作 j 也叫快指針

快指針顧名思義，走的更快，所以循環以它為基准

算法流程如下：

1. 比較 i 和 j 位置的元素是否相等
2. 如果相等，j 后移 1 位，i 不需要動
3. 如果不相等，i 后移 1 位，然后將 j 位置的值賦值到 i 新位置上，j 繼續后移一位（其實就是循環+1）
4. 重復上述過程，直到 j 指向數組最后一個元素
5. 返回 i + 1，即為新數組長度

 

復雜度分析

• 時間復雜度：O(n)，假設數組的長度是 n，那么 i 和 j 分別最多遍歷 n 步

• 空間復雜度：O(1)