Leetcode NO.27 Remove Element

最新推荐文章于 2025-09-09 20:07:09 发布

原创最新推荐文章于 2025-09-09 20:07:09 发布 · 335 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #two pointer #array #C++

Leetcode 专栏收录该内容

140 篇文章

订阅专栏

本文介绍了一种使用双指针技术高效移除数组中特定值的方法，并提供了C++实现代码。该算法通过前后两个指针配合，实现了原地删除指定元素并返回新数组长度的功能。

本题要求如下：

Given an array and a value, remove all instances of that value in place and return the new length.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

这题是道简单题，而且也不是第一次做，所以很快就做出来了。。

代码：

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int ptr_i = 0;
        int ptr_j = n - 1;
        int len = n;
        while (ptr_i < len) {
            if (A[ptr_i] == elem) {
                swap(A[ptr_i], A[ptr_j--]);
                --len;
            }
            else
                ++ptr_i;
        }
        return len;
    }
};

用的是two pointer的方法，思路就是一个前指针，一个后指针，

对前指针遍历，如果遇到了elem,则和后指针指向的元素交换，同时，后指针往前移动，len减1

此时，需要对该元素再次检测，以防其也等于elem。

由于Leetcode改了function的接口，更新下代码

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int l = 0;
        int r = nums.size() - 1;
        while (l <= r) {
            if (nums[l] == val) {
                swap(nums[l], nums[r--]);
            }
            else {
                ++l;
            }
        }
        return l;
    }
};