leetcode #27 移除元素 | 刷题之路第一站——数组类相关问题

题号 27

题目描述

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

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

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

示例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。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。

链接：leetcode #27 移除元素

解题思路1【双指针】：

1 - 首先对特殊情况进行处理：
（1）如果给定数组为空，则直接返回0；
（2）如果给定数组中所有元素都与给定值相同，则直接将给定数组清空，返回0。
2 - 对给定数组排序；
3 - 设立两个指针：
初始状态下：
target - 指向数组中第一个等于给定值的元素
i - 指向数组中最后一个等于给定值的元素的下一个元素
4 - 如果排序后数组的最后一个元素等于给定值，则表示target之前的元素均和给定值不同，返回target即可；
5 - 否则，将 nums[i] 赋值给 nums[target]，再同时将 i 和 target 后移一位，直至 i 指向数组中最后一个元素。
时间复杂度： O(nlgn)
分析：排序为O(nlgn)，元素替换为O(n)，所以整体为O(nlgn)

代码实现：

class Solution {
public:
	int removeElement(vector<int>& nums, int val) {
		sort(nums.begin(), nums.end());    //对给定数组排序
		if (nums.size() == 0) return 0;    //特殊情况处理
		if (count(nums.begin(), nums.end(), val)==nums.size())
		{
			nums.clear();
			return 0;
		}
		int target = 0;    //target指向数组中第一个等于给定值的元素
		while (target < nums.size())
		{
			if (nums[target] == val)
				break;
			else {
				target++;
			}
		}
		int i = target;    //i指向数组中最后一个等于给定值的元素的后一个元素
		while (i < nums.size()-1)
		{
			if (nums[i + 1] == nums[target])
				i++;
			else {
				i++;
				break;
			}
		}
		int a = nums.size() - 1;
		if (nums[a]==val)    //如果数组中最后一个元素与给定值相等，则返回target
		{
			return target;
		}
		else {    //否则进行元素的替换
			while (i < nums.size())
			{
				nums[target] = nums[i];
				target++;
				i++;
			}
			return target;
		}
	}
};

解题思路2 【双指针】

1 - 设立两个指针
初始状态下：
insert - 指向给定数组中第一个与给定值相等的元素
i - 指向 insert 之后第一个与给定值不等的元素
2 - 对于数组中所有的元素 nums[i] 进行如下的循环：
（1）如果 nums[i] 与给定值相同，则跳过该元素直接将 i 指向下一个元素；
（2）如果 nums[i] 与给定值不同，则将 nums[i] 的值赋值给 nums[insert]，同时将 insert 和 i 各后移一个元素。
3 - insert 的值就是数组中符合要求的元素个数，返回 insert 即可。

【观察发现，初始状态可以通过步骤2的循环实现，因此，我们只需要将两个指针值都赋值为0即可，步骤1的初始状态交由循环来实现】

【上述循环中，实现了将所有与给定值不相等的元素进行依次前移的操作，并将前面与给定值相同的元素进行了覆盖，因此可行】
备注：insert 为慢指针，i 为快指针

确认过眼神，这是我写不出来的代码 orzzzzzz
参考：leetcode 官方题解

代码实现：

class Solution {
public:
	int removeElement(vector<int>& nums, int val) {
		int insert = 0;
		for (int i = 0; i < nums.size(); i++)
		{
			if (nums[i] != val)
			{
				nums[insert] = nums[i];
				insert++;
			}
		}
		return insert;
	}
};```