Lintcode(7)-删除元素

最新推荐文章于 2020-02-18 21:29:17 发布

pengyou41

最新推荐文章于 2020-02-18 21:29:17 发布

阅读量539

点赞数

CC 4.0 BY-SA版权

分类专栏： lintcode 文章标签： lintcode

本文链接：https://blog.youkuaiyun.com/pengyou41/article/details/46679507

lintcode 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种利用C++ STL中的std::remove函数高效移除数组中特定值的方法，并提供了一个简洁的实现示例。

Q:
Remove Element.
Given an array and a value, remove all occurrences of that value in place and return the new length.
The order of elements can be changed, and the elements after the new length don’t matter.
Given an array [0,4,4,0,0,2,4,4], value=4
return 4 and front four elements of the array is [0,0,0,2]
A:
这个就是vector遍历删除的问题，可以直接用erase返回的迭代器，进行下一次访问。
这里我们不使用这种方式，直接使用std::remove，这个题目就是一行代码-:)。

class Solution {
public:
    /** 
     *@param A: A list of integers
     *@param elem: An integer
     *@return: The new length after remove
     */
    int removeElement(vector<int> &A, int elem) {
        if (!A.size()) return 0;
        A.erase(std::remove(A.begin(), A.end(), elem), A.end());
        return A.size();
    }
};