leetcode 移除元素

我在做这一题的时候用到了STL种find函数以及vector的erase函数,虽然空间复杂度满足要求,但是时间复杂度却不够优秀。代码如下,

class Solution 
{
public:
    int removeElement(vector& nums, int val)
    {
        //先处理退化的情况
        int N = nums.size();
        if (nums.size() == 0)
            return 0;
        
        int count  = 0;
        auto it = nums.begin();
        
        while(it != nums.end())
        {
            auto f_it = find(nums.begin(), nums.end(), val);

            if(f_it != nums.end())
            {
                ++count;
                auto it = nums.erase(f_it);
            }
            
            else
                break;
        }
        return N - count;
    }
};

实际上,这道题完全可以借鉴,有序向量的唯一化操作,用非指定元素覆盖指定元素,知道遍历完整整个vector,其中一人的答案如下:

class Solution {
public:

    int removeElement(vector& nums, int val) {
        int n=nums.size();
        int count=0;
        for(int i=0;i

你可能感兴趣的:(LeetCode,C++)