給定一個排序數組,你需要在 原地 刪除重復出現的元素,使得每個元素只出現一次,返回移除后數組的新長度。
不要使用額外的數組空間,你必須在 原地 修改輸入數組 並在使用 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。
你不需要考慮數組中超出新長度后面的元素。
解題思路:
解法: 雙指針
首先注意數組是有序的,那么重復的元素一定會相鄰。
要求刪除重復元素,實際上就是將不重復的元素移到數組的左側。
考慮用 2 個指針,一個在前記作 p,一個在后記作 q,算法流程如下:
1.比較 p 和 q 位置的元素是否相等。
如果相等,q 后移 1 位
如果不相等,將 q 位置的元素復制到 p+1 位置上,p 后移一位,q 后移 1 位
重復上述過程,直到 q 等於數組長度。
返回 p + 1,即為新數組長度。
畫個圖理解一下
代碼實現:
int removeDuplicates(int* nums, int numsSize){
if(numsSize == 0 || nums == NULL){
return 0;
}
int p=0;
int q=1;
while(q<numsSize){
if(nums[p] == nums[q]){
q++;
}else{
nums[p+1]=nums[q];
p++;
q++;
}
}
return p+1;
}
精簡之后的代碼如下:
int removeDuplicates(int* nums, int numsSize){
if(numsSize == 0 || nums == NULL){
return 0;
}
int p=0;
int q=1;
while(q<numsSize){
if(nums[p] !=nums[q]){
nums[p+1]=nums[q];
p++;
}
q++;
}
return p+1;
}
優化:
復雜度分析:
時間復雜度:O(n)O(n)O(n)。
空間復雜度:O(1)O(1)O(1)。
優化:
這是大部分題解都沒有提出的,在這里提一下。
考慮如下數組:
此時數組中沒有重復元素,按照上面的方法,每次比較時 nums[p] 都不等於 nums[q],因此就會將 q 指向的元素原地復制一遍,這個操作其實是不必要的。
因此我們可以添加一個小判斷,當 q - p > 1 時,才進行復制。
優化之后的代碼:
int removeDuplicates(int* nums, int numsSize){
if(numsSize == 0 || nums == NULL){
return 0;
}
int p=0;
int q=1;
while(q<numsSize){
if(nums[p] !=nums[q]){
if(q-p >1){
nums[p+1]=nums[q];
}
p++;
}
q++;
}
return p+1;
}
作者:max-LFszNScOfE
鏈接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array/solution/shuang-zhi-zhen-shan-chu-zhong-fu-xiang-dai-you-hu/
來源:力扣(LeetCode)
著作權歸作者所有。商業轉載請聯系作者獲得授權,非商業轉載請注明出處。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。