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

Posted 2021-08-07 lgz0921

题目链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/

思路：两个指针，一个表示最终结果的指针（index），一个是表示遍历的指针（i），通过判断最终结果数组中的元素是不是跟当前遍历的元素相等，来判断需不需要加进最终结果中。判断方式：nums[i] != nums[index - 2] 。这样判断的原因是，每个元素最多可以出现两次，所以与 index - 2 位置的元素比较即可。特殊情况：如果 nums.size <= 2 ，直接返回 nums.size 即可。

上代码：

class Solution {
    fun removeDuplicates(nums: IntArray): Int {
        if (nums.size <= 2) {
            return nums.size
        }
        var index = 2
        for (i in 2 until nums.size) {
            if (nums[i] != nums[index - 2]) {
                nums[index++] = nums[i]
            }
        }
        return index
    }
}