【题目】
给定一个排序数组,你需要在原地删除重复出现的元素,
使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用O(1)额外空间的条件下完成。
【示例 1】
给定数组 nums = [1,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。
你不需要考虑数组中超出新长度后面的元素。
Go:
func removeDuplicates(nums []int) int {
}
【初步思路】
- 使用一个中间变量tmp以从前往后的顺序来暂存不同元素值,初始值为nums[0]
- 通过一次循环,下标从1开始,逐个与tmp比较。
- 若不等,则更换tmp的值且往后移;若相等,则移除当前位置上元素,后面往前挪动。
- 最后计算数组里剩下的元素个数即可。
func removeDuplicates(nums []int) int {
cnt := len(nums)
var tmp = nums[0]
for i := 1; i < cnt; i++ {
// 若下标i的值不等于下标i-1 说明与前面元素不重复,则更新tmp的值,并continue
if nums[i] != tmp {
tmp = nums[i]
continue
}
// 若下标i的值等于下标i-1 说明与前面元素重复,则后面向前移动一位
nums = nums[:i]
nums = append(nums[i+1:])
}
return len(nums)
}
【反思】
问题1:不满足题目要求只使用O(1)的空间
nums = nums[:i] nums = append(nums[i+1:])
问题2:下标i上元素删掉了,后面向前进一位 =>下标i上值更新了,但再也不能拿来比较了
力扣上给这道题标注的是"简单"。。。。。。
事实上已经是把我难住了,决定找官方答案看看
【正解】
func removeDuplicates(nums []int) int {
cnt := len(nums)
if cnt <= 1 {
return cnt
}
i := 0
j := 1
for ; j < cnt; j++ {
if nums[j] == nums[i] {
continue
}
i++
nums[i] = nums[j]
}
return i + 1
}