【LeetCode-简单】27.移除元素 - 数组与双指针法

力扣题目链接

由于Leetcode中数组是用的vector,这道题可以用nums.erase(it);函数暴力破解,但要注意erase()函数在删除元素后会将位于该元素后方的剩余元素前移 ,这将导致数组长度以及后续元素下标的变化 ,删除元素后迭代器 it 不需要 it++便已经指向了下一个元素。

暴力破解代码:

cpp 复制代码
//时间复杂度:O(n^2)
//空间复杂度:O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
    	//迭代器
        for(vector<int>::iterator it = nums.begin(); it != nums.end(); ){
            if(*it == val){
                nums.erase(it);
            }else{
                it++;
            }
        }
        return nums.size();
    }
};

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int len = nums.size();
        for(int i = 0; i < len; ){
            if(nums[i] == val){
                nums.erase(nums.begin() + i);
                len--;
            }else{
                i++;
            }
        }
        return len;
    }
};

不考虑vector的因素,由于数组的元素在内存地址中是连续的,不能单独删除数组中的某个元素,只能覆盖 。本题可采用双指针 的方法。加上题目中提及元素的顺序可以改变,同向双指针相向双指针都可以使用。如果要求不能改变元素顺序,则应该使用同向双指针。

下面定义本题中的快慢指针:

  • 快指针:寻找新数组(不含有目标元素的数组)的元素 ,即用于寻找不等于val的元素
  • 慢指针:指向更新新数组下标的位置,即指向需要被覆盖的等于val的元素

同向双指针 :快慢指针同时从数组最左侧出发

  • 当快指针所指元素不等于val时,把快指针所指向的元素覆盖掉慢指针指向的元素,随后快慢指针一起后移。
  • 当快指针所指元素等于val时,仅快指针后移,此时慢指针仍指向此时数组中第一个未被覆盖掉的等于val的元素。

*当快慢指针指向同一个元素且该元素不等于val时,仍进行覆盖操作,完成覆盖后数组仍保持原样。

代码如下:

cpp 复制代码
//时间复杂度:O(n)
//空间复杂度:O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int j = 0;
        for(int i = 0; i < nums.size(); i++){
            if(nums[i] != val){
                nums[j] = nums[i];
                j++;
            }
        }
        return j;
    }
};

相向双指针 :左指针从数组最左侧 寻找等于val的元素,右指针从数组最右侧 寻找不等于val的元素。找到后,把右边不等于val的元素覆盖掉左边等于val的元素。

代码如下:

cpp 复制代码
/**
* 相向双指针方法,基于元素顺序可以改变的题目描述改变了元素相对位置,确保了移动最少元素
* 时间复杂度:O(n)
* 空间复杂度:O(1)
*/
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int leftIndex = 0;
        int rightIndex = nums.size() - 1;
        while (leftIndex <= rightIndex) {
            // 找左边等于val的元素
            while (leftIndex <= rightIndex && nums[leftIndex] != val){
                ++leftIndex;
            }
            // 找右边不等于val的元素
            while (leftIndex <= rightIndex && nums[rightIndex] == val) {
                -- rightIndex;
            }
            // 将右边不等于val的元素覆盖左边等于val的元素
            if (leftIndex < rightIndex) {
                nums[leftIndex++] = nums[rightIndex--];
            }
        }
        return leftIndex;   // leftIndex一定指向了最终数组末尾的下一个元素
    }
};
相关推荐
AC使者8 分钟前
#B1630. 数字走向4
算法
冠位观测者12 分钟前
【Leetcode 每日一题】2545. 根据第 K 场考试的分数排序
数据结构·算法·leetcode
古希腊掌管学习的神1 小时前
[搜广推]王树森推荐系统笔记——曝光过滤 & Bloom Filter
算法·推荐算法
qystca1 小时前
洛谷 P1706 全排列问题 C语言
算法
古希腊掌管学习的神1 小时前
[LeetCode-Python版]相向双指针——611. 有效三角形的个数
开发语言·python·leetcode
浊酒南街1 小时前
决策树(理论知识1)
算法·决策树·机器学习
就爱学编程1 小时前
重生之我在异世界学编程之C语言小项目:通讯录
c语言·开发语言·数据结构·算法
学术头条1 小时前
清华、智谱团队:探索 RLHF 的 scaling laws
人工智能·深度学习·算法·机器学习·语言模型·计算语言学
Schwertlilien2 小时前
图像处理-Ch4-频率域处理
算法
IT猿手2 小时前
最新高性能多目标优化算法:多目标麋鹿优化算法(MOEHO)求解TP1-TP10及工程应用---盘式制动器设计,提供完整MATLAB代码
开发语言·深度学习·算法·机器学习·matlab·多目标算法