移除元素——快慢指针解法理解

最新推荐文章于 2025-05-20 18:28:23 发布

apex-l

最新推荐文章于 2025-05-20 18:28:23 发布

阅读量187

点赞数 2

分类专栏： leecode刷题记录文章标签： leetcode 算法数据结构

本文链接：https://blog.youkuaiyun.com/weixin_52764246/article/details/143953247

版权

leecode刷题记录专栏收录该内容

1 篇文章

订阅专栏

题目链接:https://leetcode.cn/problems/remove-element/description/

题目描述

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

暴力解法

遍历数组，假设在i位置遇到遇到与val相同的数据，则将剩下的数组元素向前一步覆盖num[i]=nums[i+1]，直到遍历结束。

快慢指针法

这里的指针并不是真正的指针，是对数组下标的记录。快指针(fast）与慢指针(slow），快指针始终快于慢指针。
我们可以把遍历的过程理解成买菜的过程，从菜篮子里挑选合格菜品放进塑料袋里。nums是蔬菜筐里的蔬菜，经过我挑选过的蔬菜个数是fast，slow代表挑选好放进塑料袋里的菜品数量。

首先fast挑选蔬菜，合格（num[fast]!=val）则放进菜篮子(slow)里，完成塑料袋里的菜品加以nums[slow]=nums[fast],塑料袋里蔬菜的数量slow++,经我手挑选过的菜的数量fast++；不合格（num[fast]==val）的由fast舍弃掉fast++，不往塑料袋里装。直到我挑完菜筐里的所有菜(fast==nums.size()-1）为止。

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

上述挑选菜品的过程中，合格的菜品经过遍历全部入袋，不合格的则直接舍弃，由于fast始终快于slow，所以使用nums[slow]=nums[fast]不会影响nums[fast]的判断，相比于暴力解法，相当于在fast遍历的过程中完成了数组元素的移位。