LeetCode刷题之数组 第26题 删除排序数组中的重复项

题目描述

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

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

方法一：暴力解法

思路

老规矩，先来一遍暴力解法。与 27 题移除元素类似，这道题的暴力解法是两层 for 循环，第一层遍历数组，第二层更新数组，如果要删除某个元素，则该元素后面的所有元素都要进行前移操作

代码

class Solution {
    public int removeDuplicates(int[] nums) {
        int length = nums.length;
        for (int i = 1; i < length; ++i) {
            if (nums[i] == nums[i - 1]) {
                for (int j = i - 1; j < length - 1; ++j) {
                    nums[j] = nums[j + 1];
                }
                --length;
                --i;             // 注意 --i 要放在 if 语句中，只有 if 语句成立的时候才需要进行 i--
            }
        }
        return length;
    }
}

复杂度分析

时间复杂度：O(n^2)
空间复杂度：O(1)

方法二：双指针法

思路

保留两个 i 和 j，i 是满指针，j 是快指针，当 nums[j] 与给 nums[i] 相等时，递增 j 以跳过该元素，否则就将 nums[j] 复制到 nums[i]，并同时递增两个指针。重复这一过程，直到 j 到达数组的末尾。此时数组的长度为 i。

代码

class Solution {
    public int removeDuplicates(int[] nums) {
        int i = 0;
        int length = nums.length;
        for (int j = i + 1; j < length; ++j) {
            if (nums[j] != nums[i]) {
                ++i;
                nums[i] = nums[j];
            }
        }
        return i + 1;
    }
}

复杂度分析

时间复杂度：O(n), 假设数组的长度是 n，那么 i 和 j 分别最多遍历 n 步。
空间复杂度：O(1)。