leetcode27之移除元素

题目描述：

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

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

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

示例 1:

给定 nums = [3,2,2,3], val = 3,

函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-element

代码实现：

def removeitem(nums, tar):
    ans = 0
    for num in range(len(nums)):
        if nums[num] != tar:
            nums[ans] = nums[num]
            ans += 1
        else:
            continue

    return ans


print("---------------测试removeitem()-------------")
nums = [3, 3, 3, 3, 3]
target = 2
res = removeitem(nums, target)
print("res=", res)


def removeitem1(nums, tar):
    '''

    :param nums:
    :param tar:
    :return:
    '''
    for i in range(len(nums) - 1, -1, -1):
        if (nums[i] == tar):
            nums.pop(i)

    return len(nums)


print("---------------测试removeitem1()-------------")
nums = [3, 3, 3, 3, 3]
target = 2
res = removeitem1(nums, target)
print("res=", res)


def removeitem2(nums, tar):
    '''

    :param nums:
    :param tar:
    :return:
    '''
    l=len(nums)
    for i in range(l):
        if (nums[i] != tar):
            nums.append(nums[i])
            # nums.pop(i)
    del nums[:l]
    return len(nums)

print("---------------测试removeitem2()-------------")
nums = [3, 3, 3, 3, 3]
target = 2
res = removeitem2(nums, target)
print("res=", res)

输出：

---------------测试removeitem()-------------
res= 5
---------------测试removeitem1()-------------
res= 5
---------------测试removeitem2()-------------
res= 5

总结：1、原地移除元素，考虑使用双指针。方法一正是采用了类似双指针策略来实现。遍历整个数组，若该元素与目标元素相等，跳过继续循环，若不想等，执行nums[ans]=nums[num]操作，ans+=1。返回ans即为最终数组长度。

2、方法二思路很值得借鉴，考虑从数组末尾开始遍历，如果相等，弹出该元素；如果不相等，继续循环。返回为len(nums)

3、方法三追加元素+删除数组方法。从头开始遍历数组，如果相等，跳过继续循环；如果不等，在数组末尾追加该元素。循环结束后删除原有长度数组，返回len(nums)。该方法依然满足O(1)复杂度要求