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
}