2.移除元素

题目

给你一个数组nums和一个值val，你需要原地移除所有数值等于val的元素。元素的顺序可能发生改变。然后返回nums中与val不同的元素的数量。

假设nums中不等于val的元素数量为k，要通过此题，您需要执行以下操作:

更改nums数组，使nums的前k个元素包含不等于val的元素。nums的其余元素和nums的大小并不重要。 返回k。

示例 1：

输入：nums = [3,2,2,3], val = 3 输出：2, nums = [2,2,,] 解释：你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。 你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。 示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2 输出：5, nums = [0,1,4,0,3,,,_] 解释：你的函数应该返回 k = 5，并且 nums 中的前五个元素为 0,0,1,3,4。 注意这五个元素可以任意顺序返回。 你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

思路

双指针：

• 用一个变量k记录不等于val的元素数量，从0开始
• 从头开始遍历nums中的元素，与val进行比较，如果不等于val，那就将nums[k]的值设置为nums中当前的元素，同时后移k

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

class Solution {
    fun removeElement(nums: IntArray, `val`: Int): Int {
        var k = 0
        for (num in nums) {
            if (num != `val`) {
                nums[k] = num
                k++
            }
        }
        return k
    }
}

双指针优化

上面的方案存在一个问题，就是例如数组为[1, 2, 3, 4, 5]，而val为1时。我们需要把每一个元素都左移一位。

注意到题目上说：元素的顺序可以改变。

实际上我们只需要将最后一个元素5移动到序列开头，取代元素1，得到序列[5, 2, 3, 4]就可以。

思路： - 还是用双指针，一个从前往后left，一个从后往前right - 从前往后的指针left的判断方式还是如同上面的思路 - 如果left上的值等于val，那就把left位置的值换成right位置的值，同时移动right继续循环 - 直到left > right相等，那就都遍历完了

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        left = 0
        right = len(nums) - 1
        while left <= right:
            if (nums[left] == val):
                nums[left] = nums[right]
                right -= 1
            else:
                left += 1
        return left   

class Solution {
    fun removeElement(nums: IntArray, `val`: Int): Int {
        var left = 0
        var right = nums.size - 1
        while (left <= right) {
            if (nums[left] == `val`) {
                nums[left] = nums[right]
                right --
            } else {
                left ++
            }
        }

        return left
    }
}

• 下一篇

• 邮箱 ：charon.chui@gmail.com
• Good Luck!