LeetCode 27 移除元素

题目:

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

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

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

解题:

定义两个指针,不同的就赋值,相同的就跳过

代码:


func removeElement(nums []int, val int) int {
    l := len(nums)
    if l == 0 {
        return 0
    }
    j := 0
    for i := 0; i < l; i++ {
        if nums[i] != val {
            nums[j] = nums[i]
            j++
        }
    }
    return j
}

官方解答: 

// 双指针
func removeElement(nums []int, val int) int {
    left := 0
    for _, v := range nums { // v 即 nums[right]
        if v != val {
            nums[left] = v
            left++
        }
    }
    return left
}
// 双指针优化
func removeElement(nums []int, val int) int {
    left, right := 0, len(nums)
    for left < right {
        if nums[left] == val {
            nums[left] = nums[right-1]
            right--
        } else {
            left++
        }
    }
    return left
}

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注