問題描述

給定一個(gè)數(shù)組和一個(gè)值，在這個(gè)數(shù)組中原地移除指定值和返回移除后新的數(shù)組長度。

不要為其他數(shù)組分配額外空間，你必須使用 O(1) 的額外內(nèi)存原地修改這個(gè)輸入數(shù)組。

元素的順序可以改變。超過返回的新的數(shù)組長度以外的數(shù)據(jù)無論是什么都沒關(guān)系。

示例:

給定 nums = [3,2,2,3]，val = 3，

你的函數(shù)應(yīng)該返回 長度 = 2，數(shù)組的前兩個(gè)元素是 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