Leetcode - Pointers/Array - Remove Element

最新推荐文章于 2025-08-07 20:34:46 发布

原创最新推荐文章于 2025-08-07 20:34:46 发布 · 496 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

30 篇文章

订阅专栏

本文介绍了一种在数组中移除指定值并返回新长度的算法实现。该算法通过双指针技术，有效地将目标值从数组中移除，并允许元素顺序变化。文章提供了两种不同的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) {
        if(n==0)
            return 0;
        int *p=&A[0];
        int *q=&A[n-1];
        while(q>=p)
        {
            if(*p==elem)
            {
                if(*q==elem)
                {
                    q--;
                }
                else
                {
                    *p=*q;
                    q--;
                    p++;
                }
                n--;
            }
            else
            {
                p++;
            }
        }
        return n;
    }
};

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        if(A==NULL)
            return 0;
        for(int i=0;i<n;)
        {
            if(A[i]==elem)
            {
                swap(A[i],A[n-1]);
                n--;
            }
            else
                i++;
        }
        return n;
    }
};