26. Remove Duplicates from Sorted Array

题目描述和难度

  • 题目描述：

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

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

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

说明:

为什么返回数值是整数，但输出的答案是数组呢?

请注意，输入数组是以“引用”方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝
int len = removeDuplicates(nums);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

思路分析

求解关键：简单来说，得记录一下当前遍历的元素之前的那个元素，用作比较。

应该充分利用排好序的数组这个特性来完成。 注意：应该注意到一些特殊的测试用例，例如 nums = [] 的时候。注意，题目要求返回新数组的长度。 编码：

参考解答

参考解答1

import java.util.Arrays;

// 常规题目：这里利用到数组的有序性，如果遇到和上一个一样的元素，就什么都不做
public class Solution {

    // [1,1,2,3,4,4]
    // [1,2,3,4,4]
    public int removeDuplicates(int[] nums) {

        int len = nums.length;
        if (len == 0) {
            return 0;
        }
        int pre = nums[0];
        int j = 0;
        for (int i = 1; i < len; i++) {
            if (nums[i] != pre) {
                pre = nums[i];
                j++;
                nums[j] = nums[i];
            }
        }
        // 注意 j 是遍历到的与之前不同元素的个数，要把第 1 个元素算进去，所以要加 1
        return j + 1;
    }

    public static void main(String[] args) {
        int[] nums = {1, 1, 2, 2, 2, 3, 3, 4, 4, 4};
        int ret = new Solution().removeDuplicates(nums);
        System.out.println(ret);
        System.out.println(Arrays.toString(nums));
    }
}

参考解答2

import java.util.Arrays;

public class Solution2 {

    public int removeDuplicates(int[] nums) {
        int len = nums.length;
        if (len == 0) {
            return 0;
        }
        int j = 0;
        int i = 1;
        for (; i < len; i++) {
            if (nums[i] == nums[j]) {
                // 什么都不做
            } else {
                nums[++j] = nums[i];
            }
        }
        // 返回的是数组的长度，所以要 + 1
        return j + 1;
    }
}

本篇文章的地址为 https://liweiwei1419.github.io/leetcode-solution/leetcode-0026-remove-duplicates-from-sorted-array ，如果我的题解有错误，或者您有更好的解法，欢迎您告诉我 liweiwei1419@gmail.com 。