【LeetCode】27. 移除元素

示例 1:

给定 nums = [3,2,2,3], val = 3,

函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time: 2018/7/3
# @Author: xfLi
# The file...

def removeElement(nums, val):
    """
    :type nums: List[int]
    :type val: int
    :rtype: int
    """
    i = 0
    while i < len(nums):
        if nums[i] == val:
            nums.pop(i)
        else:
            i += 1
    return len(nums),nums

if __name__ == '__main__':
    nums = [2, 3, 3, 2, 3]
    result, s = removeElement(nums, 3)
    print(result, s)