【LeetCode刷题记录】简单篇-27-移除元素

【题目描述】

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

---

【测试用例】

示例1：

        输入：nums = [3,2,2,3]

        输出：2，nums = [2,2]

        解释：函数应该返回新的长度2，并且 nums 中的前两个元素均为2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为2，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

示例2：

        输入：nums = [0,1,2,2,3,0,4,2]

        输出：5，nums = [0,1,3,0,4]

---

【思路分析】

这道题和LeetCode简单题的26题异曲同工，不同之处在于循环遍历数组nums时候内部的判断条件不同和pos1初始位置、后移条件不同。具体的可以参考我的26题题解并结合本题代码。

---

【参考代码】

C实现

#include <stdio.h>

//easy-27-移除元素
int removeElement(int* nums, int numsSize, int val);

int main(){
	int nums[] = {3,2,2,3};
	int numsSize = sizeof(nums)/sizeof(nums[0]);
	int val = 3;
	int k = removeElement(nums, numsSize, val);
	int i;
	printf("%d\n", k);
	for(i=0;i<k;++i){
		printf("%d ",nums[i]);
	}
	return 0;
}

int removeElement(int* nums, int numsSize, int val) {
    int pos1=0, pos2=0;
    while(pos2<numsSize){
    	if(nums[pos2] == val){
    		pos2++;
		}else{
			nums[pos1++] = nums[pos2++];
		}
	}
	return pos1;
} 

C++实现

#include <iostream>
#include <vector>
using namespace std;

//easy-27-移除元素
class Solution {
	public:
	    int removeElement(vector<int>& nums, int val);
};

int Solution::removeElement(vector<int>& nums, int val){
	int pos1=0, pos2=0;
	while(pos2<nums.size()){
		if(nums[pos2] == val){
			pos2++;
		}else{
			nums[pos1++] = nums[pos2++];
		}
	}
	return pos1;
} 

int main(){
	vector<int> nums = {3,2,2,3};
	int val = 3;
	Solution sol;
	int k = sol.removeElement(nums, val);
	int i;
	cout<<k<<endl;
	for(i=0;i<k;++i){
		cout<<nums[i]<<" ";
	}
	return 0;
}

---