leetcode 探索 数组和字符串 移除元素

题目

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。

想法

由于是要原地移除，如果单纯的找到一个就把后面的元素往前移动，这将是O(n*n)的解法。题目里说了元素的顺序可以改变，并且数组可以超出新长度后面的元素。所以我们可以考虑把要删除的元素置换到数组的后面去，要删除一个那么数组的长度就会减一，所以替换到数组的后面去，这个元素就相当于删除了。但是我们不能直接一次替换就移动到下一个元素，因为替换的时候有可能把要删除的元素替换过来。所以替换一次之后，继续判断，直到遇到的是不等的。

另外一个想法，标记清楚，先标记，然后之后统一的清除。

解法

func removeElement(nums []int, val int) int {
    length := len(nums)
    var j = 0
    for i := 0; i < length - j; {
        if nums[i] == val {
            nums[i], nums[length-j-1] = nums[length-j-1], nums[i]
            // fmt.Printf("i, last: %d, %d, %v", i, length-j-1, nums)
            j++
        } else {
            i++
        }
    }
    
    
    return len(nums) - j
}