Leetcode-80. Remove Duplicates from Sorted Array II

公众号每天发布一篇关于Leetcode解题技巧的文章，尝试从多角度、不同方法对题目进行解析。欢迎大家关注！

题目描述（中等难度）

26 题的升级版，给定一个数组，每个数字只允许出现 2 次，将满足条件的数字全部移到前边，并且返回有多少数字。例如 [ 1, 1, 1, 2, 2, 3, 4, 4, 4, 4 ]，要变为 [ 1, 1, 2, 2, 3, 4, 4 …] 剩余部分的数字不要求。

解法一 快慢指针

利用26 题的思想，慢指针指向满足条件的数字的末尾，快指针遍历原数组。并且用一个变量记录当前末尾数字出现了几次，防止超过两次。

package Remove_Duplicates_from_Sorted_Array_II;

public class Remove_Duplicates_from_Sorted_Array_II {
	
	public static int removeDuplcates(int[] nums) {
		int slow=0;
		int count=1;
		int max=2;
		for(int fast=1;fast<nums.length;fast++) {
			if(nums[fast]==nums[slow]) {
				if(count<max) {
					slow++;
					nums[slow]=nums[fast];
					count++;
				}
			}else {
				slow++;
				nums[slow]=nums[fast];
				count=1;
				
			}
		}
		return slow+1;
	}
	
	public static void main(String args[]) {
		int[]nums= {1,1,2,2,2,3,3,4,4,4,4};
		int ans=removeDuplcates(nums);
		System.out.println(ans);
	}
}

时间复杂度：O（n）。

空间复杂度：O（1）。

解法二

解法一中，我们用一个变量 count 记录了末尾数字出现了几次。而由于给定的数组是有序的，我们可以更直接。将当前快指针遍历的数字和慢指针指向的数字的前一个数字比较（也就是满足条件的倒数第 2 个数），如果相等，因为有序，所有倒数第 1 个数字和倒数第 2 个数字都等于当前数字，再添加就超过 2 个了，所有不添加，如果不相等，那么就添加。s 代表 slow，f 代表 fast。

//相等的情况
1 1 1 1 1 2 2 3
  ^   ^ 
  s   f
//不相等的情况  
1 1 1 1 1 2 2 3
  ^       ^ 
  s       f
public int removeDuplicates2(int[] nums) {
    int slow = 1;
    int fast = 2;
    int max = 2;
    for (; fast < nums.length; fast++) {
        //不相等的话就添加
        if (nums[fast] != nums[slow - max + 1]) {
            slow++;
            nums[slow] = nums[fast];
        }
    }
    return slow + 1;
}

时间复杂度：O（n）。

空间复杂度：O（1）。

python解法

 i = 0
 j = 0
count = 1
while(j<=len(nums)-2):
    j += 1
    if nums[j] == nums[i]:
        count += 1
        if count > 2:
            del nums[i]
            j = i
            count = 1
    else:
        i = j
        count = 1

总结

快慢指针是个好东西，解法二直接利用有序，和倒数第 n 个比，从而保证末尾的数字不超过 n 个，太强了。