LeetCode第26题删除有序数组中的重复项RemoveDuplicatesFromSortedArray

Q26RemoveDuplicatesFromSortedArray

题目描述

给你一个升序排列的数组 nums ，请你原地删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。元素的相对顺序应该保持 一致 。

由于在某些语言中不能改变数组的长度，所以必须将结果放在数组nums的第一部分。更规范地说，如果在删除重复项之后有k个元素，那么 nums 的前k个元素应该保存最终结果。

将最终结果插入nums的前k个位置后返回k。

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

判题标准:

系统会用下面的代码来测试你的题解:

int[] nums = [...]; // 输入数组
int[] expectedNums = [...]; // 长度正确的期望答案
int k = removeDuplicates(nums); // 调用
assert k == expectedNums.length;
for (int i = 0; i < k; i++) {
assert nums[i] == expectedNums[i];
}
如果所有断言都通过，那么您的题解将被 通过。

示例 1：

输入：nums = [1,1,2]
输出：2, nums = [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]
解释：函数应该返回新的长度 5 ， 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。

https://leetcode.cn/problems/remove-duplicates-from-sorted-array

---

解题思路

题目要求是原地修改数组，即不能构造新数组，且时间为O(1)
考虑到数组是有序的，即相等元素的下标一定是连续的，可以考虑使用双指针来解题。
定义fast和slow两个指针，fast指向当前遍历到的元素，slow指针指向构造好的数组的最后一个位置，两个指针从0开始。fast指针从0遍历到len-1，对于遍历到的每个元素，判断nums[fast] 与 nums[slow] 是否一致，如果不一致则表明当前元素为新的不同元素，将其放入 nums[++slow] 中，slow仍指向构造好的数组最后一个位置，最后将slow加一为数组长度返回。

代码

public class Q26RemoveDuplicatesFromSortedArray {
    public int removeDuplicates(int[] nums) {
        if(nums==null || nums.length==0){
            return 0;
        }
        int len = nums.length;
        int fast = 0;
        int low = 0;
        while(fast < len){
            if(nums[fast] != nums[low]){
                nums[++low] = nums[fast];
            }
            fast++;
        }
        return ++low;
    }
}

本地测试

public class Q26RemoveDuplicatesFromSortedArrayTest {

    private Q26RemoveDuplicatesFromSortedArray q = new Q26RemoveDuplicatesFromSortedArray();

    @Test
    public void test1() {
        int[] test = new int[]{1,1,2};
        assertEquals(2, q.removeDuplicates(test));
        assertArrayEquals(new int[]{1,2,2}, test);
    }

    @Test
    public void test2() {
        int[] test = new int[]{0,0,1,1,1,2,2,3,3,4};
        assertEquals(5, q.removeDuplicates(test));
        assertArrayEquals(new int[]{0,1,2,3,4,2,2,3,3,4}, test);
    }
}

提交