【面试经典150题】LeetCode80·删除有序数组中的重复项2·【Java】

题目

给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使得出现次数超过两次的元素只出现两次 ，返回删除后数组的新长度。

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

说明：

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

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

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

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

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

示例 1：

输入：nums = [1,1,1,2,2,3]
输出：5, nums = [1,1,2,2,3]
解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3。 不需要考虑数组中超出新长度后面的元素。
示例 2：

输入：nums = [0,0,1,1,1,1,2,3,3]
输出：7, nums = [0,0,1,1,2,3,3]
解释：函数应返回新长度 length = 7, 并且原数组的前七个元素被修改为 0, 0, 1, 1, 2, 3, 3。不需要考虑数组中超出新长度后面的元素。

提示：

1 <= nums.length <= 3 * 10的4次方
-10的4次方 <= nums[i] <= 10的4次方
nums 已按升序排列

详解

精简题目：

• 题目给定一个数组，我们要把重复的第三个删除掉，并且返回剩余数的个数

思路详解：

• 题目要求不能创造新数组，空间复杂度有要求，但是时间复杂度没有要求
• 所以，初步想的是用两个for，记录好前两个相同的数字，从第三个相同数字的位置后的元素依次向前挪
• 这样的时间效率极其低下
• 容易发现，在后面的元素，在达到最终结果的位置，要挪多次才能达到
• 所以我们要使每个元素一步到位，就能挪到最终位置
• 这时就可以用双指针，前面的指针是进行向前推进元素，后面的指针是存储数字
• 但是由于题目上对空间复杂度有要求，所以我们需要对原数组进行双指针操作

代码

初版(时间效率比较低下的双循环)：

class Solution {
    public int removeDuplicates(int[] nums) {
        int n = nums.length;
        int ans = 0;
        int temp=0;
        for(int i=0;i<n-1;i++){
            if(nums[i]==nums[i+1]){
                temp++;
                if(temp==2){//遇到两次以上的，后面的依次向前挪
                    temp=1;
                    for(int j=i+1;j<n-1;j++){
                        nums[j]=nums[j+1];
                    }
                    n--;
                    i--;
                }
                else{
                    ans++;
                }
            }
            else{
                ans++;
                temp=0;
            }
        }
        return ans+1;
    }
}

改进后（时间效率比较高的双指针）：

class Solution {
    public int removeDuplicates(int[] nums) {
        int n=nums.length;
        int ans=2;
        for(int i=2;i<n;i++){
            if(nums[ans-2]!=nums[i]){
                nums[ans++]=nums[i];
            }
        }
        return Math.min(ans,n);//防止原数组不超过2
    }
}