Leetcode_remove-element

最新推荐文章于 2024-03-25 22:40:48 发布

原创最新推荐文章于 2024-03-25 22:40:48 发布 · 529 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

155 篇文章

订阅专栏

本文介绍了一种解决LeetCode上移除指定元素问题的方法。通过先对数组进行排序，然后查找并移除所有目标元素，从而实现高效地更新数组长度。此方法不保留原数组中元素的相对顺序。

地址：http://oj.leetcode.com/problems/remove-element/

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.

思路：给定一个数组A及其长度n，求其删掉元素elem后的长度和元素，题目不要求数组元素的相对位置，故考虑排序。排序后遍历，如果存在elem，那么elem必定相邻(如果有多个的话)，找到所有的elem后开始移动元素。

参考代码：

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int begin = 0, diff = 0;
        std::sort(A, A+n);
        for(int i = 0; i < n; ++i)
        {
            if(A[i]==elem)
            {
                begin = i;
                ++diff;
                while(++i < n && A[i]==elem)
                {
                    ++diff;
                }
                if(i<n)
                {
                    for(;i<n;++i)
                    {
                        A[i-diff]=A[i];
                    }
                }
            }
        }
        return n-diff;
    }
};