【一天一道Leetcode】移除元素

本篇推文共计2000个字，阅读时间约3分钟。

01

题目描述

题目描述：

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

不要使用额外的数组空间，你必须仅使用O(1)额外空间并原地修改输入数组。

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

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 
并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，
也会被视作正确答案。

提示：

1. 0 <= nums.length <= 100

2. 0 <= nums[i] <= 50

3. 0 <= val <= 100

02

思路和方法

由题意最后只用输出新数组的长度

我们可以利用遍历从数组第一个元素进行遍历，依次对比当前数组元素是否与给定的目标值相同，将不同的数组元素往数组前面移动。

类似于筛选，将符合条件的元素，排到队伍前面。

最后输出与给定目标值元素不同的那部分数组长度。

例如我们用一个例子来直观地说明。

初始数组num=[1,1,2,2,3,3]，给定的值为3

设定1个起点进行遍历
i=0，

设定1个节点，
node=0 
用于把和给定目标值不相同的数组元素往数组前面移动，

如果num[i]!=val
则令num[node]=num[i]

利用这样的方式，将同一个数组与给定目标值不同的元素都往前移动。

我们的代码输出为：

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        node=0
        for i in range(0,len(nums)):
            if nums[i]!=val:
                nums[node]=nums[i]
                node+=1
            i+=1
        return node

往期回顾

【年终总结】你好2021，再见2020。

【秋招纪实录】一篇特别正经的【腾讯】求职经验分享

【一天一道Leetcode】笨阶乘

☆ END ☆

你与世界

只差一个

公众号