2020.05.26 力扣 - 数组 - 26 删除排序数组中的重复项

124 阅读2分钟

【题目】

给定一个排序数组,你需要在原地删除重复出现的元素,

使得每个元素只出现一次,返回移除后数组的新长度。

不要使用额外的数组空间,你必须在原地修改输入数组并在使用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
}