[LeetCode] Remove Element

最新推荐文章于 2024-02-23 09:08:20 发布

原创最新推荐文章于 2024-02-23 09:08:20 发布 · 287 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

131 篇文章

订阅专栏

本文介绍了一种高效的数组元素移除算法，通过双指针技巧实现对指定元素的快速删除，并保留非目标元素的位置不变。提供了两种实现方式，一种适用于待移除元素较多的情况，另一种则通用。

//设置两个指针，i用来定位elem，j用来赋值给i
class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int cnt = 0,i = 0,j = n - 1;
        while(i <= j){
            while(A[j] == elem && j > 0 && j != i){//用非elem元素赋值给A[i],j != i 防止重复计算
                cnt ++;
                j --;
            }
            if(A[i] == elem){
                cnt ++;
                A[i] = A[j];
                if(j > 0)
                    j --;
            }
            i ++;
        }
        return n - cnt;
    }
};

方法2：若remove的元素多，则赋值次数少，remove的元素少，则赋值次数多，不管怎样，方法1的赋值操作是最少的。

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int k = 0;
        for(int i = 0;i < n;i ++){
            if(A[i] != elem)
                A[k ++] = A[i];
        }
        return k;
    }
};