码迷,mamicode.com
首页 > 其他好文 > 详细

Remove Element

时间:2017-03-12 16:45:53      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:替代   函数   ++   index   erase   turn   move   问题   不同   

思路:直接使用index记录新vector的下标即可,不过还是应该注意便捷条件的处理

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        if(nums.size() == 0)
            return 0;
        
        int index = 0;
        for(int i=0; i<nums.size(); ++i)
        {
            if(nums[i] != val)
                nums[index++] = nums[i];
        }
        
        return index;
    }
};

 这个问题也可以使用stl中distance和remove两个函数实现,这里需要注意remove函数不是直接删除vector中的元素,而是用后面的元素替代前面的元素,所以在这个问题中直接返回nums.size()是不对的,其值是不变的

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        return distance(nums.begin(), remove(nums.begin(), nums.end(), val));
    }
};

 但是使用erase函数可以直接抹去vector中的元素,这也是两个函数的不同之处,可以看出remove函数返回的是最后一个值的迭代器

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        nums.erase(remove(nums.begin(), nums.end(), val), nums.end());
        return nums.size();
    }
};

 

Remove Element

标签:替代   函数   ++   index   erase   turn   move   问题   不同   

原文地址:http://www.cnblogs.com/chengyuz/p/6537992.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!