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

LeetCode 27. 移除元素

大梦三千秋
关注TA
已关注
手记 173
粉丝 7
获赞 168

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。

注意这五个元素可为任意顺序。

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

解题思路


思路:双指针

审题,本题的要求是【不要使用额外的数组空间】。

定义双指针,front ,tail。当 nums[front] 的值与给定的 val 的值相等时,则 front 指针向前移动,跳过该元素。若是不相等,则将 nums[front] 的值复制给 nums[tail],然后两个指针向前同时移动。

循环重复这个过程,但 front 到达末尾时,返回的新数组长度即为 tail。

具体过程看以下图解:

图解

具体代码如下。

代码实现


class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
		# 定义双指针
        front = 0
        tail = 0
		
		# 循环遍历数组
        while front < len(nums):
	        # 当 front 指针所对应的值与给定 val 值不同时
	        # front 指针向前移动,跳过该元素
            if nums[front] != val:
            	# 将 front 所对应的值复制给 tail 指针所对应的值
                nums[tail] = nums[front]
                tail += 1
            front += 1

        return tail

实现结果


实现结果


以上就是使用双指针的方法,移动指针时,判断元素值是否与给定的值相同,决定是否跳过该元素。若不同则将前指针的值复制给后指针对应的值,否则跳过该元素,重复该过程,从而解决《27. 移除元素》问题的主要内容。


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