双指针—删除有序数组中的重复项

删除有序数组中的重复项

1. 只允许重复一个

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

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

说明:

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

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

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

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

在函数里修改输入数组对于调用者是可见的。
根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。

for(int i = 0; i < len; i++) {
print(nums[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 。不需要考虑数组中超出新长度后面的元素。

提示：

0 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按升序排列

public int removeDuplicates(int[] nums) {
        if(nums.length<2) {  return nums.length; } 
			int i=0 ; //快慢指针,i指针更新元素，j指针遍历
			for (int j=1; j < nums.length; j++) {
				if(	nums[j] !=	 nums[i]) {//如果是新元素直接加
					i++;
					nums[i]	 =  nums[j];
				} 
			}
			return i+1;
    }

时间复杂度：O(n)
空间复杂度：O(1)

2. 允许重复两个
删除有序数组中的重复项 II
给你一个有序数组 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 * 104
-104 <= nums[i] <= 104
nums 已按升序排列。

	public static void main(String[] args) {
		int[] nums= {1,1,1,2,2,3,3,5,5};
		int len = removeDuplicates(nums);
		for (int i = 0; i < len; i++) {
		    System.out.println(nums[i]);
		}
	}
			
	public static int removeDuplicates(int[] nums) {
		if(nums.length<3) {    return nums.length;		} 
		int i=1 , j=2; //快慢指针,i指针更新元素，j指针遍历
		while( j	<	nums.length	) {
			if(	nums[j] !=	 nums[i]) {//如果是新元素直接加
				i++;
				nums[i]	 =  nums[j];
				j++;
			}else {//如果是旧元素，就要考虑符合条件否
				if(nums[j]  !=  nums[i-1]) {//不超过两个
					i++;
					nums[i]  =  nums[j];
				}
				j++;
			}
		}
	   return i+1;
	}

时间复杂度：O(n)
空间复杂度：O(1)

Leetcode官方解题方法：

class Solution {
    public int removeDuplicates(int[] nums) {
        int n = nums.length;
        if (n <= 2) {
            return n;
        }
        int slow = 2, fast = 2;
        while (fast < n) {
            if (nums[slow - 2] != nums[fast]) {
                nums[slow] = nums[fast];
                ++slow;
            }
            ++fast;
        }
        return slow;
    }
}

class Solution {
    public int removeDuplicates(int[] nums) {   
        return process(nums, 2);
    }
    int process(int[] nums, int k) {
        int u = 0; 
        for (int x : nums) {
            if (u < k || nums[u - k] != x) nums[u++] = x;
        }
        return u;
    }
}

时间复杂度：O(n)
空间复杂度：O(1)

总结：

• 我们定义两个指针 slow 和 fast 分别为慢指针和快指针，其中慢指针表示处理出的数组的长度，快指针表示已经检查过的数组的长度

• 特别地，数组的前两个数必然可以被保留，因此对于长度不超过2 的数组，我们无需进行任何处理，对于长度超过 2的数组，我们直接将双指针的初始值设为 2即可

• 由于是保留 k 个相同数字，对于前 k 个数字，我们可以直接保留

• 对于后面的任意数字，能够保留的前提是：与当前写入的位置前面的第 k 个元素进行比较，不相同则保留

就有了通用解法——双指针解法：

class Solution {
    public int removeDuplicates(int[] nums) {   
        return process(nums, 1);
    }
    int process(int[] nums, int k) {
        int u = 0; 
        for (int x : nums) {
            if (u < k || nums[u - k] != x) nums[u++] = x;
        }
        return u;
    }
}

时间复杂度：O(n)
空间复杂度：O(1)