【每日一题】27.移除元素

最新推荐文章于 2024-11-03 20:43:31 发布

ahojcn

最新推荐文章于 2024-11-03 20:43:31 发布

阅读量194

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法

转载请注明出处

本文链接：https://blog.csdn.net/Hanoi_ahoj/article/details/105354674

版权

LeetCode 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

leet-code 27. 移除元素 https://leetcode-cn.com/problems/remove-element/

解法1 暴力

C++ 实现要注意迭代器失效的问题！
4 ms 6.3 MB

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        auto it = nums.begin();
        while (it != nums.end()) {
            if (*it == val) {
                it = nums.erase(it);
            } else {
                ++it;
            }
        }
        return nums.size();
    }
};

解法2 双指针法

类似于 26.删除数组中的重复项

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        size_t s = 0, f = 0;
        while (f < nums.size()) {
            if (nums[f] != val) {
                nums[s++] = nums[f];
            }
            f++;
        }
        return s;
    }
};

解法3 双指针法优化

这个解法是看了题解里面的。
双指针法，当 nums 中只有较少的元素 == val 的时候。方法2中，会将一些元素做不必要的移位操作。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        size_t n = nums.size();
        size_t i = 0;
        while (i < n) {
            if (nums[i] == val) {
                nums[i] = nums[n - 1];  // 和最后一个有效值交换
                --n;  // 有效元素 - 1
            } else {
                ++i;
            }
        }
        return n;
    }
};