之前做过删除元素的题,当时简单地使用了erase函数而没有探究底层原因。原来erase函数删除是O(N)复杂度,并没有领悟题目的意思。
今天重新做了原地删除元素的题以及原地删除有序数组重复项的题,用了双指针+一层for循环解决,快指针指向新数组所需元素,慢指针指向新数组的下标。快指针获取的值赋给慢指针所在位置,慢指针从0开始。最后慢指针的大小代表数组的大小。
AC题解,使用上述双指针的思路,将慢指针指向第0个,默认留下该数字。快指针在这里要指向第一个,因为要留下一个的数字而不是去除该数字,所以从后一个开始判断。当快指针指向的一个数字与当前(即慢指针)留下的数字不相同,则将慢指针指向后面一位,将快指针的值赋给慢指针,然后慢指针往后一位。最后返回的数组大小为slow+1。因为slow为数组下标,指向新数组的最后一位,得到数组大小要加一。
class Solution {
public int removeDuplicates(int[] nums) {
int slow = 0;
if(nums.length > 1){
for(int fast = 1;fast < nums.length;fast++){
if(nums[fast]!=nums[slow]){
slow++;
nums[slow] = nums[fast];
}
}
}
return slow+1;
}
}