删除排序数组中的重复项（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)
        }
    }

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

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