27. 移除元素 @Python @Go

原题

https://leetcode.cn/problems/remove-element/description/

思路

双指针

复杂度

时间：O(n)
空间：O(n)

Python代码

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        ans = 0
        l = 0
        for r in range(len(nums)):
            if nums[r] == val:
                continue
            nums[l] = nums[r]
            l += 1
            ans += 1
        return ans

Go代码

func removeElement(nums []int, val int) int {
	ans := 0
	l := 0
	for r := 0; r < len(nums); r++ {
		if nums[r] == val {
			continue
		}
		nums[l] = nums[r]
		l++
		ans++
	}
	return ans
}