Question

我正在尝试为给定的数组和值编写程序，删除该值的所有实例并返回新的长度。

实施例：给定输入数组nums = [3,2,2,3]，val = 3

它应返回length = 2，其中nums的前两个元素为2。

这是我的代码：

代码1：

def removeElement(self, nums, val):
    """
    :type nums: List[int]
    :type val: int
    :rtype: int
    """
    i = 0
    j = len(nums) - 1

    while i <= j:
        while i <= j and nums[j] != val:
            j -= 1

        while i <= j and nums[i] == val:
            i += 1

        if i <= j:
            nums[i], nums[j] = nums[j], nums[i]

    return len(nums[i:])

以相反的顺序返回数组切片。

Input:
[3,2,2,3]
3 
Output: [3,3] 
Expected: [2,2]

但是，如果我在代码1的末尾稍作修改，它会给我正确的输出：

    nums[:] = nums[i:]
    return len(nums[i:])

代码2：

def removeElement(self, nums, val):
    """
    :type nums: List[int]
    :type val: int
    :rtype: int
    """
    i = 0
    j = len(nums) - 1

    while i <= j:
        while i <= j and nums[j] != val:
            j -= 1

        while i <= j and nums[i] == val:
            i += 1

        if i <= j:
            nums[i], nums[j] = nums[j], nums[i]

    nums[:] = nums[i:]
    return len(nums)

我无法弄清楚为什么我的代码1无效。有人可以帮我理解为什么切片不能按预期工作吗？

Answer 1

这可以达到您的目的（＆＃34; ...删除该值的所有实例并返回新的长度＆＃34;）：

def remove_element(nums, val):
    nums[:] = [x for x in nums if x != val]
    return len(nums)

测试：

nums = [3, 2, 2, 3]
val = 3
print(remove_element(nums, val))
print(nums)

输出：

2
[2, 2]

Answer 2

你的第一个例子有效。

切片时，会创建新列表。因此，在您的第一个代码示例中，您在最后创建一个包含正确的结果的新列表，但从不返回它。

在第二个代码示例中，您将新创建的列表分配给原始列表，因此可以访问最终结果。

Python数组切片无法按预期工作

2 个答案: