leetcode27.移除元素

1.题目
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
2.示例
示例1:
给定 nums = [3,2,2,3], val = 3,函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。
示例2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
3.思路
基本思路是遍历数组,遇到满足条件的数时,和数组最后一个值交换。当交换时当前遍历的位置与数组有效的位置相同时需要提前结束运算,否则继续遍历。
4.代码
//有效数组开始位置0,结束位置n。
int removeElement(vector< int>& nums, int val) {
int i=0,n=nums.size()-1;
for(;i<=n;i++){
if(nums[i]==val){
swap(nums[i],nums[n]);
n–;
i–;//避免nums[n]= =val;
}
}
return n+1;
}

猜你喜欢

转载自blog.csdn.net/qq_14962179/article/details/85543113