欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LintCode 题目:删除元素

程序员文章站 2022-03-24 21:45:31
...

URL:https://www.lintcode.com/problem/remove-element/description

描述

给定一个数组和一个值,在原地删除与值相同的数字,返回新数组的长度。

元素的顺序可以改变,并且对新的数组不会有影响。

样例

Example 1:
	Input: [], value = 0
	Output: 0


Example 2:
	Input:  [0,4,4,0,0,2,4,4], value = 4
	Output: 4
	
	Explanation: 
	the array after remove is [0,0,0,2]

 

1.失败

在代码段中添加:

for(vector<int>::iterator it = A.begin(); it!=A.end(); it++) {
		    if(*it==elem){
                A.erase(it);
            }
	    }
        return A.size();

得到:

LintCode 题目:删除元素

删除的不干净,还有剩余,原因未知。

 

2.成功

在代码段中添加:

vector<int> B;
        for(vector<int>::iterator it = A.begin(); it!=A.end(); it++) {
		    if(*it!=elem){
                B.push_back(*it);
            }
	    }
	    A.clear();
	    A=B;

即可:

LintCode 题目:删除元素