鏈接
https://leetcode-cn.com/problems/remove-element/description/
要求
給定一個數(shù)組 nums 和一個值 val,你需要原地**移除所有數(shù)值等于 *val *的元素,返回移除后數(shù)組的新長度。
不要使用額外的數(shù)組空間,你必須在原地修改輸入數(shù)組并在使用 O(1) 額外空間的條件下完成。
元素的順序可以改變。你不需要考慮數(shù)組中超出新長度后面的元素。
給定 nums = [3,2,2,3], val = 3,
函數(shù)應(yīng)該返回新的長度 2, 并且 nums 中的前兩個元素均為 2。
給定 nums = [0,1,2,2,3,0,4,2], val = 2,
函數(shù)應(yīng)該返回新的長度 5, 并且 nums 中的前五個元素為 0, 1, 3, 0, 4。
相關(guān)代碼
class Solution(object):
def removeElement(self, nums, val):
while val in nums:
nums.remove(val)
return len(nums)