Given an array and a value, remove all instances of that value in place and return the new length.
Do not allocate extra space for another array, you must do this in
place with constant memory.The order of elements can be changed. It doesn’t matter what you leave
beyond the new length.Example: Given input array nums = [3,2,2,3], val = 3
Your function should return length = 2, with the first two elements of
nums being 2.
方法一:
很多习惯性的都是想写新数组,但这题要求不增加新内存。所以只有在原数组基础改写,基本就是覆盖原数组来写新数组。
int removeElement(vector<int>& nums, int val) {
int n = nums.size();
if (n == 0) return 0;
int count = 0;
for (int i = 0; i + count < n; ) {
if (nums[i + count] == val) {
count++;
continue;
}
nums[i] = nums[i + count];
i++;
}
return n-count;
}
我们用count来统计有多少个重复的val,然后考虑延后几个取数来写。很简单的代码。但其实可以思路更直白,写的更简单。如下:
int removeElement(vector<int>& nums, int val) {
int num=0,i;
int n = nums.size();
for(i=0;i<n;++i)
if(nums[i] != val) nums[num++]=nums[i];
return num;
}
也就是依次检索,不是val就是写入。是就跳过。
方法二:
也就是hint里提到的,如果想要删除的元素很少,大量的assignment工作都是在写入需要保留的元素。我们可以利用不用保持原有顺序这一个特点来操作。
int removeElement(vector<int>& nums, int val) {
int i;
int n = nums.size();
for(i=0; i<n; ) {
if(nums[i] == val) {
nums[i]=nums[n-1];
n--;
}
else i++;
}
return n;
}
利用改变顺序这个特点,不停的将要删除的元素移动到尾部,然后同时减少数组长度,相当于只用对要删除的元素操作,在该元素rare的情况下大大减少了assignment的操作。