LeetCode—移除元素

前言

题目：给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。题目链接

有如下解法，其中一些解法不满足题目要求的会注明，写出来只是提供一个思路想法。如有不对，请指正！

1.暴力求解

遍历整个数组，每当遇到一个等于val的元素时，就将其从数组中删除。直到遍历完整个数组，返回新的数组长度。

代码实现：

//暴力求解
int removeElement(int* nums, int numsSize, int val) {
	int i = 0;
	while (i < numsSize)
	{
		if (nums[i] == val)
		{
			for (int j = i; j < numsSize - 1; j++) //把后面的元素往前移动进行覆盖，就相当于删除val元素
			{
				nums[j] = nums[j + 1];
			}
		}
		else
			i++;
	}
	return numsSize; //返回新数组的长度
}

时间复杂度为$O(n^2)$，空间复杂度为$O(1)$。
满足题目要求，但是效率太低。

2.创建临时的空间

创建一个临时数组，将原数组中不等于val的元素复制到临时数组中，最后将临时数组复制回原数组中。

代码实现：

//创建临时空间
int removeElement(int* nums, int numsSize, int val) {
	int* tmp = (int*)malloc(sizeof(int) * numsSize);
	if (tmp == NULL) //这个判断其实也可以省略，在本题中。
	{
		perror("malloc fail\n");
		return;
	}
	int count = 0;
	int i = 0;
	for (i = 0; i < numsSize; i++)
	{
		if (nums[i] != val)
		{
			tmp[count] = nums[i];
			count++;
		}
	}
	for (i = 0; i < count; i++)
	{
		nums[i] = tmp[i]; 
	}
	free(tmp);//释放临时空间，这题中可以不用把tmp置为NULL
	return count;
}

本题中释放了临时数组的内存后，不需要再使用tmp指针，也不需要将其置为NULL。因为在函数结束时，这个指针就会被销毁，不再存在，因此不会出现悬空指针问题。
时间复杂度为$O(n)$，空间复杂度为$O(n)$。
相较于第一种虽然效率提高了，但空间复杂度不满足题目要求。

3.双指针法

双指针法的基本思路是维护一个慢指针和一个快指针，快指针用于遍历整个数组，慢指针用于指向需要保留的元素的位置。当快指针遇到需要删除的元素时，快指针继续向前遍历，慢指针不变，直到遇到一个需要保留的元素时，将其复制到慢指针的位置并将慢指针向前移动一位。

代码实现：

int removeElement(int* nums, int numsSize, int val){
    int src=0;
    int dest=0;
    while(src<numsSize)
    {
        if(nums[src]!=val)
        {
            nums[dest++]=nums[src++];
        }
        else
        {
            src++;
        }
    }
    return dest; //返回长度
}

时间复杂度为$O(n)$，空间复杂度为$O(1)$。
可见最好的解法是双指针法，不仅时间效率高，也节省空间！

今天的算法分享就到这里啦！