删除有序数组重复原元素

题目描述

给你一个 非严格递增排列 的数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。

考虑 nums 的唯一元素的数量为 k ，你需要做以下事情确保你的题解可以被通过：

• 更改数组 nums ，使 nums 的前 k 个元素包含唯一元素，并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums的大小不重要。
• 返回 k 。

示例 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 。不需要考虑数组中超出新长度后面的元素。

提示：

• 1 <= nums.length <= 3 * 104

• -104 <= nums[i] <= 104

• nums 已按 非严格递增 排列

题解

使用零时数组 + hash map

func removeDuplicates(nums []int) int {
	elementMap := map[int]bool{}
	tmpArr := []int{}
	for _, num := range nums {
		_, found := elementMap[num]
		if found {
			continue
		}
		tmpArr = append(tmpArr, num)
		elementMap[num] = true
	}
	for i, _ := range tmpArr {
		nums[i] = tmpArr[i]
	}
	return len(tmpArr)
}

时间复杂度: 2n

空间复杂度: 3n

原地操作，降低时间空间复杂度

func removeDuplicates(nums []int) int {
	slow := 0
	fast := slow + 1
	for fast < len(nums) {
		if nums[slow] == nums[fast] {
			fast++
		} else {
			nums[slow+1] = nums[fast]
			slow++
			fast++
		}
	}
	return slow + 1
}

时间复杂度: n

空间复杂度: 1

总结

这题比较简单，数组基本操作。

使用了知识点:

• 双指针