删除排序数组中的重复项(LeetCode-26)

写在前面:看到很多程序员朋友都在刷LeetCode,刚好最近在学习swift以及数据结构算法,所以用swift刷LeetCode应该是个很好的练习方式。自己会在playground上编写代码及测试,并上传github持续更新,地址:LeetCode-Swift

题目描述

给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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。

你不需要考虑数组中超出新长度后面的元素。

审题

  • 有序可变数组,返回删除重复元素后的数组的长度
  • 不使用额外的数组空间,即不能创建新的数组变量
  • 最后可变数组变为一个不含重复元素的有序数组

代码

func removeDuplicates(_ nums: inout [Int]) -> Int {
    
    if nums.count <= 1 {
        return nums.count
    }
    
    var i = 0
    for j in 0..<nums.count {
        if nums[j] != nums[i] {
            i += 1
            nums[i] = nums[j]
        }
    }
    
    if i != nums.count {
        for _ in (i+1) ..< nums.count {
            nums.remove(at: i+1)
        }
    }
    
    return i + 1;
}

测试

var nums0 = [0]
removeDuplicates(&nums0)   // 1
print(nums0)               // [0] 

var nums1 = [0,1,1,2,2,2,3,3,4]
removeDuplicates(&nums1)  // 5
print(nums1)              //[0,1,2,3,4]

说明

1、题目要求输入参数可以被修改,所以用了inout关键字(输入输出参数)
2、nums为指针变量,函数调用时记得前面加上&(手写别忘了)
3、题目要求数组要原地删除重复的元素,也就是如果存在重复元素,函数调用结束后输入数组应该有变化,所以应该有下面这段代码:

 if i != nums.count {
        for _ in (i+1) ..< nums.count {
            nums.remove(at: i+1)
        }
    }

网上一些解答都没有这步处理,我觉得是不对的。

好了,这道题就到这,有什么疑问欢迎交流讨论~

    原文作者:Flum_X
    原文地址: https://www.jianshu.com/p/3d69974dd5cd
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞