0026-刪除有序陣列中的重複項

LanceZh發表於2024-06-11

26. 刪除有序陣列中的重複項

給你一個 非嚴格遞增排列 的陣列 nums ,請你 原地 刪除重複出現的元素,使每個元素 只出現一次 ,返回刪除後陣列的新長度。元素的 相對順序 應該保持 一致 。然後返回 nums 中唯一元素的個數。

考慮 nums 的唯一元素的數量為 k ,你需要做以下事情確保你的題解可以被透過:

更改陣列 nums ,使 nums 的前 k 個元素包含唯一元素,並按照它們最初在 nums 中出現的順序排列。nums 的其餘元素與 nums 的大小不重要。
返回 k 。
判題標準:

系統會用下面的程式碼來測試你的題解:

int[] nums = [...]; // 輸入陣列
int[] expectedNums = [...]; // 長度正確的期望答案

int k = removeDuplicates(nums); // 呼叫

assert k == expectedNums.length;
for (int i = 0; i < k; i++) {
    assert nums[i] == expectedNums[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 。不需要考慮陣列中超出新長度後面的元素。
 

提示:

1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按 非嚴格遞增 排列

解題思路

雙指標,左側指標 l 及其左邊表示已經是有序的,右側指標r 查詢比左側指標 nums[l] 大的元素,放入 nums[l+1] 的位置。


func removeDuplicates(nums []int) int {

    n := len(nums)
    if n <= 1 {
        return n
    }
    // 左右邊界.
    // 小於等於 l 的位置是已經有序的,結束時返回 l+1 即可。
    l, r := 0, 1
    
    for l < r && r < n {
        // 找到 nums[r] > nums[l]
        if nums[r] > nums[l] {
            // l 左側是有序的,所以要將大於 nums[l] 的元素放在 nums[l+1] 的位置,
            // 並且移動 l 
            l++
            nums[l] = nums[r]
        }
        // 無論找到與否都要移動 r
        r++
    }
    
    // l 是索引,要返回數量,所以要加 1
    return l+1
}

相關文章