LeetCode Remove Element

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

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

0 ·

CC 4.0 BY-SA版权

面试题专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种在数组中移除特定值并返回新长度的算法实现。该方法通过覆盖目标值来完成移除操作，并允许改变数组内元素的顺序。

题目：

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 k = 0;
        for(int i = 0; i < n-k; i++) {
            if(A[i] == elem) {
                if(i == n-1-k) {
                    k++;
                    break;
                }
                A[i] = A[n-1-k];
                k++;
                i--;
            }
        }
        return n-k;
    }
};