首页 > 其他 > 详细

Remove Element

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

思路:直接使用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

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!