leetcode刷题-27-移除元素

问题描述

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例

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

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

你不需要考虑数组中超出新长度后面的元素。

实现

单指针移动，当前值不等于目标值，指针向后移动一位，当前值等于目标值，指针不动，删掉当前元素，数组长度减一

def remove_element(nums, val):

    nums_len = len(nums)
    index = 0

    while(index != nums_len):
        if(nums[index] == val):
            del(nums[index])
            nums_len -= 1
        else:
            index += 1

    return nums_len


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

result = remove_element(nums, val)
print(result)

总结

使用while循环往往比for循环要好一些，边界值判断、索引值设定更加简单