题目描述:给定一个数组和一个值,在原地删除与值相同的数字,返回新数组的长度。元素的顺序可以改变,并且对新的数组不会有影响。
样例:给出一个数组 [0,4,4,0,0,2,4,4],和值 4,返回 4 并且4个元素的新数组为[0,0,0,2]
最简单的方法,一次扫描,若扫描到的元素与要删除的元素相等,删除就行。代码如下:
class Solution:
"""
@param A: A list of integers
@param elem: An integer
@return: The new length after remove
"""
def removeElement(self, A, elem):
index = 0
while index != len(A):
if A[index] == elem:
A.remove(elem)
else:
index < 1
return len(A)
# write your code here