27. 移除元素
给定一个数组和一个值,在这个数组中原地移除指定值和返回移除后新的数组长度。
不要为其他数组分配额外空间,你必须使用 O(1) 的额外内存原地修改这个输入数组。
元素的顺序可以改变。超过返回的新的数组长度以外的数据无论是什么都没关系。
示例:
给定 nums = [3,2,2,3],val = 3, 你的函数应该返回 长度 = 2,数组的前两个元素是 2。
思路:
这题和前面的26题基本一样,可以套用,但是用while的写法更合适。
答案:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
if(nums.size()==0) return 0;
int i=0,p=0;
while(i<nums.size()&&p<nums.size()){
if(nums[p]-val) p++; //搜查坏人
else{
if(i<=p) i=p+1; //修正i的值,从p后面寻找好人
while(i<nums.size()){
if(nums[i]-val){
nums[p]=nums[i]; //好人取代坏人的位置
nums[i]=val;
p++; //当前坏人已消除,搜查下一个坏人
i++;
break; //跳出内部while循环,上面i不加也可以,但是会多做一次不必要的比较
}
i++;
}
}
}
return p;
}
};
反思:
这个代码写得有点不太易懂了,因为提交的时候中间错了两次,所以修改了一些细节,然后顺利通过了。但是以后应该尽量避免写出这种需要注释的代码。
但是这个代码运行结果还是不错的:
不过虽然同样是4ms,但人家给出的范例明显比我的要优雅高端得多:
还没仔细看这个代码,下次有时间再看,今天先到这里,歇了。