天机阁

26. 删除有序数组中的重复项

2022-07-01 · 3 min read
LeetCode

题目

给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。

由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。

将最终结果插入 nums 的前 k 个位置后返回 k 。

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

难度:🌟

示例 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 已按 升序 排列

题解

这题主要考察指针的用法,定义一个指针指向数组第一个元素,将数组中元素与指针挨个比较,如果不同,如果相同则进行删除操作且数组中元素向前前进一位。如果相同,则将指针指向新的元素。

代码实现

func removeDuplicates(nums []int) int {
	n := len(nums)
	elem := nums[0]
	elemIdx := 0
	for i := 1; i < n; i++ {
		if elem == nums[i] { // 需要删除 i 位置上的元素
			for j := i; j < n-1; j++ {
				nums[j] = nums[j+1] // 删除了 nums[i] 位置的元素
			}
			n-- // 元素数量-1
			i = elemIdx
			// 有需要删除的元素时 elem 不需要更新
		} else { // 更新 elem
			elem = nums[i]
			elemIdx = i
		}
	}
	return n
}

// 快慢指针的思路实现
func removeDuplicates(nums []int) int {
	// 使用快慢指针,慢指针走在后面,快指针走在前面探路,找到一个不重复的元素就赋值给慢指针,并让慢指针前进一步
	// 这样就保证了 nums[0...slow]都是无重复元素,当快指针遍历完整个数组后,程序结束
	slow, fast := 0, 0
	for fast < len(nums) {
		if nums[fast] != nums[slow] {
			slow++
			nums[slow] = nums[fast]
		}
		fast++
	}
	return slow + 1
}