数组-27. 移除元素

原创已于 2024-05-03 10:08:24 修改 · 101 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #数据结构

于 2024-05-03 09:57:58 首次发布

可信考试C++/C 专栏收录该内容

24 篇文章

订阅专栏

本文介绍了如何在力扣LeetCode平台上实现一个高效算法，原地移除数组中所有等于给定值的元素，同时保持O(1)额外空间复杂度。给出了两种方法：暴力求解和使用快慢指针优化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

. - 力扣（LeetCode）

给你一个数组 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。

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

1.暴力求解

class Solution {
public:
    // 3,2,2,3
    int removeElement(vector<int>& nums, int val) {
        int returnLen = nums.size();
        int idx = 0;
        int len = returnLen;
        //3,2,2,3
        //    i
        for (int i = len - 1; i >= 0; i--) {
            if (val == nums[i]) {
                returnLen--;
                for (int j = i; j < len - 1; j++) {
                    nums[j] = nums[j + 1];
                }
            }
        }

        return returnLen;
    }
};

2.使用快慢指针

定义快慢指针

快指针：寻找新数组的元素，新数组就是不含有目标元素的数组
慢指针：指向更新新数组下标的位置

class Solution {
public:
    // 3,2,2,3
    int removeElement(vector<int>& nums, int val) {
        int slowIdx = 0, fastIdx = 0;
        int size = nums.size();
        int len = size;
        while (slowIdx < len && fastIdx < len) {
            if (nums[fastIdx] == val) {
                size--;
            } else {
                nums[slowIdx] = nums[fastIdx];
                slowIdx++;
            }
            fastIdx++;
        }

        return size;
    }
};