問題描述:
給定一個排序數組,你需要在原地刪除重複出現的元素,使得每個元素只出現一次,返回移除後數組的新長度。
不要使用額外的數組空間,你必須在原地修改輸入數組並在使用 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
。 你不需要考慮數組中超出新長度後面的元素。
基本思路:
雙指針,一個遍歷數組,一個指向下一個插入的位置。
AC代碼:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() == 0) return 0;
int j = 0; // 遍歷整個nums數組
int i = 0; // 獲得下一個插入的地方
int k = nums[0]; // 作爲比較的值
while (j < nums.size()) {
if (nums[j] != nums[i]) {
++i;
k = nums[j];
nums[i] = k;
}
++j;
}
return i + 1;
}
};
其他經驗:
在取nums[0]的時候注意nums爲空的情況。