問題描述
給定一個數組和一個值,在這個數組中原地移除指定值和返回移除后新的數組長度。
不要為其他數組分配額外空間,你必須使用 O(1) 的額外內存原地修改這個輸入數組。
元素的順序可以改變。超過返回的新的數組長度以外的數據無論是什么都沒關系。
示例:
給定 nums = [3,2,2,3],val = 3,
你的函數應該返回 長度 = 2,數組的前兩個元素是 2。
解答方法
class Solution:
def removeElement(self, nums, val):
"""
:type nums: List[int]
:type val: int
:rtype: int
"""
a=0
for i in nums:
if i != val:
nums[a]=i
a=a+1
return a