天天看点

LeetCode----Remove Element

Remove Element

Given an array and a value, remove all instances of that value in place and return the new length.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

分析:

移除数组中给定的元素,返回之后数组的长度。简单的遍历+交换。

代码:

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        right = len(nums) - 1
        i = 0
        while i <= right:
            if nums[i] == val:
                nums[i], nums[right] = nums[right], nums[i]
                right -= 1
            else:
                i += 1
        return right + 1