LeetCode刷题篇—27.移除元素

题目

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

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

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

思路

一、一次遍历法
题目要求输出的是更新后的数组长度，而且可以修改输入数组，那么可以考虑将所有数值等于val的元素全部移到数组尾端，然后输出前部分的长度。
（1）定义新的变量n等于数组的长度；
（2）遍历数组，若元素等于val，那么就将该元素移到数组尾端，即nums[n-1]，并n–，反之，则进行下次循环。
（3）最终输出n。
二、双指针法
（1）定义左右两个指针 l和r，分别从数组的两端开始遍历，当左指针小于右指针时一直进行循环；
（2）当左指针小于右指针且左指针指向的元素值不等于val时，左指针向右继续移动（即为了找到第一个等于val的元素）
（3）当左指针小于右指针且右指针指向元素值等于val时，右指针向左继续移动（即为了找到第一个不等于val的元素）
（4）交换此时左右指针指向的元素，即将等于val的元素换到了数组后部分；
（5）最后左右指针相等，如果此时指向的元素等于val，则返回 l，反之返回 l+1 。

求解方法

int removeElement(vector<int>& nums, int val) {//一次遍历法
		int i = 0;
		int n = nums.size();
		while (i < n)
		{
			if (nums[i] == val)
			{
				nums[i] = nums[n - 1];
				n--;//这个n--就很像队列或者链表中的弹出操作
			}
			else
			{
				i++;
			}
		}
		return n;
	}

int removeElement1(vector<int>& nums, int val) {//双指针法
		int l = 0, r = nums.size()-1;
		while (l < r)
		{
			while ((l < r) && (nums[l] != val))
			{
				l++;
			}
				
			while ((l < r) && (nums[r] == val))
			{
				r--;
			}
			swap(nums[l],nums[r]);
			if (l == r)
			{
				return nums[l] == val ? l : l + 1;
			}
		}
	}