leetcode27 Remove Element 去掉指定的元素

最新推荐文章于 2024-01-20 09:00:07 发布

原创最新推荐文章于 2024-01-20 09:00:07 发布 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode #算法

python 同时被 2 个专栏收录

68 篇文章

订阅专栏

leetcode

8 篇文章

订阅专栏

本文介绍了一种在不使用额外空间的情况下，从列表中移除指定值并返回新长度的算法。通过双指针技巧，该算法实现了原地修改列表的目标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array and a value, remove all instances of that value in place and return the new length.

Do not allocate extra space for another array, you must do this in place with constant memory.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

Example:
Given input array nums = [3,2,2,3], val = 3

Your function should return length = 2, with the first two elements of nums being 2.

给定一个列表和一个值，去掉列表中的这个值并且返回新的长度（即列表中不等于这个值的个数）。

不允许分配额外内存给一个新的列表。

元素的顺序可以被改变。

举例：

给的列表为[3,2,2,3]，值为3。

你的函数应该然后2，并且列表的前两个元素都是2.

解法：

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        start, end = 0, len(nums) - 1
        while start <= end:
            if nums[start] == val:
                nums[start], nums[end] = nums[end], nums[start]
                end -= 1
            else:
                start += 1
        return len(nums[:start])


print Solution().removeElement([2, 3, 3, 2], 3)

思路：

start为列表头，end为列表尾。

从头开始循环，如果start指针所在元素等于值，将start指针所在元素与end指针所在元素互换，并且end指针-1，如果不等于则start指针+1。

while循环，当start指针大于end指针时，结束循环，并且返回结果。