https://leetcode.cn/problems/remove-element/ 点击跳转到LeetCode平台OJ页面(27.移除元素)
题目:
👉 示例1:
输入:nums = [3,2,2,3], val = 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], val = 2 输出:5, nums = [0,1,4,0,3]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
思想1:暴力解法
解题思路:
1. 从前往后遍历nums,找到val第一次出现的位置
2. 将val之后的所有元素整体往前搬移,即删除该val
3. nums中有效元素个数减少一个
时间复杂度:O(n²) 空间复杂度:O(1)
代码实现:
int removeElement(int* nums, int numsSize, int val){ int i = 0; for(i = 0;i < numsSize;i++) { if(nums[i] == val) { int j = 0; for(j = i + 1;j < numsSize;j++) { nums[j - 1] = nums[j]; } numsSize--;//等于val的元素被覆盖掉了,数组元素就减去1 i--;//下标i以后的数值向前移动了一位,所以i需要回到原来的位置继续寻找 } } return numsSize; }
思想2:创建一个temp数组
解题思路:
1. 创建一个长度与nums相同的数组temp
2. 遍历nums,将nums中所有与val不同的元素搬移到temp中
3. 将temp中所有元素拷贝回nums中
时间复杂度:O(n) 空间复杂度:O(n)
代码实现:
int removeElement(int* nums, int numsSize, int val) { // 1. 申请numSize个元素的新空间 int* temp = (int*)malloc(sizeof(int) * numsSize); if (NULL == temp) { return 0; } // 2. 将nums中非val的元素搬移到temp中---尾插到temp中 int count = 0; for (int i = 0; i < numsSize; ++i) { if (nums[i] != val) { temp[count] = nums[i]; ++count; } } // 3. 将temp中删除val之后的所有元素拷贝到nums中 memcpy(nums, temp, sizeof(int) * count); free(temp); //temp = NULL; return count; }
思想3:双指针
解题思路:
双指针的方法只需要一个循环即可解决,时间复杂度为 O(n) 。
先定义一个fast快指针(用于寻找不同于val的元素)和一个slow慢指针(用于存储更新的值;)
说明:1.这里的快慢指针本质想象为利用索引下标访问数组。
2.该题条件是快指针在小于所有元素的情况下进行查找。
快指针用于寻找不等于val的元素,当遇到不等于val的值时,快指针将此时指向的值赋值给慢指针此时指向的值,随后两个指针继续向后挪动,快指针当遇到等于val的值时,快指针直接跳过此值继续寻找,慢指针此时不动。最后只需输出慢指针的索引即可。
时间复杂度:O(n) 空间复杂度:O(1)
代码实现:
int removeElement(int* nums, int numsSize, int val){ int slow = 0; int fast = 0; while(fast < numsSize) { if(nums[fast] != val) { nums[slow++] = nums[fast++]; } else fast++; } return slow; }