去除已排序数组中的重复元素

2016-09-16 craneyuan 更多博文 » 博客 » GitHub »

algorithm LeetCode

原文链接 https://crane-yuan.github.io/2016/09/16/Remove-sorted-array-duplicated-value/
注:以下为加速网络访问所做的原文缓存,经过重新格式化,可能存在格式方面的问题,或偶有遗漏信息,请以原文为准。


题目描述

给定一个已排序的数组,去除数组中的重复元素,只保留一个重复的元素,并且返回新的数组长度。

要求: 不要给数组分配额外的空间,你必须使用常量的内存大小进行原地操作。

例如: 给出数组A=[1,1,2],你的函数调用之后必须返回长度length=2,并且A现在变成[1,2]。

输入

一个已排序的数组,例如[1,1,2]。

输出

返回数组新的长度,例如length=2。

快慢指针法

设置fast指针遍历数组,slow指针指向不重复元素的下一位。

<!--more-->

public static int removeDuplicates(int[] nums)
{
    if (nums.length < 1)
        return nums.length;
    int slow = 1;
    for (int fast = 1; fast < nums.length; fast++) {
        if (nums[fast] != nums[slow - 1]) {
            nums[slow++] = nums[fast];
        }
    }
    return slow;
}

动画演示:

动画演示

扩展

去除已排序数组中的重复元素,保留指定位数。

public static int removeDuplicatesN(int[] nums, int repeatN)
{
    if (nums.length <= repeatN)
        return nums.length;
    int index = repeatN;
    for (int i = repeatN; i < nums.length; i++) {
        if (nums[i] != nums[index - repeatN]) {
            nums[index++] = nums[i];
        }
    }
    return index;
}

参考文章