【腾讯精选练习50题】26. 删除有序数组中的重复项（简单）

题目描述

给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。

不要使用额外的数组空间，你必须在“原地”修改输入数组 并在使用 O(1) 额外空间的条件下完成。

• 0 <= nums.length <= 3 * 104
• -$10^4$ <= nums[i] <= $10^4$
• nums 已按升序排列

示例1

输入：nums = [1,1,2]

输出：2, nums = [1,2]

解释：函数应该返回新的长度 2 ，并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出新长度后面的元素。

示例2

输入：nums = [0,0,1,1,1,2,2,3,3,4]

输出：5, nums = [0,1,2,3,4]

解释：函数应该返回新的长度 5 ， 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。

思路分析

• 这里提出了一个要求“O(1)空间”。值的存储只能依赖于数组本身
• 这里考虑使用双指针。p指向不重复的位置，i遍历数组。当遇到不重复时，将下标为i的数值放到p+1的位置。同时p右移。这样就不需要交换位置，只需要把不重复的值，覆盖在重复上。

AC代码

// 执行用时：8 ms, 在所有 Go 提交中击败了87.25%的用户
// 内存消耗：4.5 MB, 在所有 Go 提交中击败了72.95%的用户
func removeDuplicates(nums []int) int {
    if len(nums) <= 1 {
        return len(nums)
    }
    p := 0
    for i := 1; i < len(nums); i++ {
        if nums[i] != nums[p] {
            p = p + 1
            nums[p] = nums[i]
        }
    }
    return p + 1
}

总结

• 这道题是第二次做了，第一次其实没明白O(1)额外空间。做过就知道了。所以要多做多学。