继续浏览精彩内容
慕课网APP
程序员的梦工厂
打开
继续
感谢您的支持,我会继续努力的
赞赏金额会直接到老师账户
将二维码发送给自己后长按识别
微信支付
支付宝支付

27. 移除元素

海绵宝宝撒
关注TA
已关注
手记 417
粉丝 40
获赞 124

27. 移除元素

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

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

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

示例 1:

给定nums=[3,2,2,3],val=3,函数应该返回新的长度2, 并且nums 中的前两个元素均为2。你不需要考虑数组中超出新长度后面的元素。

示例 2:

给定nums=[0,1,2,2,3,0,4,2],val=2,函数应该返回新的长度5, 并且nums 中的前五个元素为0,1,3,0,4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

说明:

为什么返回数值是整数,但输出的答案是数组呢?

请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

//nums是以“引用”方式传递的。也就是说,不对实参作任何拷贝int len = removeElement(nums, val);// 在函数里修改输入数组对于调用者是可见的。// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。for (int i = 0; i < len; i++) {    print(nums[i]);}

解题思想:

利用一个while 循环获取数组中的值跟 val 进行比较, 相同删除,因为删除后原数组长度(lenght)将会 -1,删除位置后面的数据会前移一位,所以下标 i 不需要 +1

代码:

class Solution:

    def removeElement(self, nums, val):

        """

        :type nums: List[int]

        :type val: int

        :rtype: int

        """

        i=0

        lenght = len(nums)

        while i < lenght:

            if nums[i] == val:

                nums.pop(i)

                lenght-=1

                i-=1

            i+=1

        return lenght


webp



作者:不爱去冒险的少年y
链接:https://www.jianshu.com/p/434b17088c37


打开App,阅读手记
0人推荐
发表评论
随时随地看视频慕课网APP