1

题目描述

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

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。


给定数组 nums = [1,1,2], 

函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。

解题思路

采用双指针方式处理。
  1. 初始指针 i, j 位置为 0
  2. 指针 j 先向右移动
  3. 比较指针 i,j 元素是否相同,若指针i,j元素相同则指针 j 继续向右移动;反之将指针 j 元素复制到指针 i + 1 处元素
  4. 当指针 j 移动到数组末尾时则停止

其流程如下图所示:

实现

public static int solution (int[] nums) {
    int i = 0, j = 0;

    while (true) {
        // 指针 j 向右移动
        j++;

        // 指针 j 移动到数组末尾则退出,说明数组元素都判断了去重
        if (j >= nums.length) {
            break;
        }

        if (nums[j] == nums[i]) {
            // 指针 i, j 元素相同,说明重复元素;
            // 指针 j 继续向右移动
            continue;
        } else {
            // 指针 i, j 元素不相同;则将指针 j 元素复制到 指针 i 后一位,这样就保证指针 i 后元素不重复
            nums[i + 1] = nums[j];
            // 指针 i 向右移动,继续处理
            i++;
        }
    }

    return i + 1;
}

断风雨
11 声望3 粉丝