题目描述(中等难度)
26 题的升级版,给定一个数组,每个数字只允许出现 2 次,将满足条件的数字全部移到前边,并且返回有多少数字。例如 [ 1, 1, 1, 2, 2, 3, 4, 4, 4, 4 ],要变为 [ 1, 1, 2, 2, 3, 4, 4 ...] 剩余部分的数字不要求。
解法一 快慢指针
利用26 题的思想,慢指针指向满足条件的数字的末尾,快指针遍历原数组。并且用一个变量记录当前末尾数字出现了几次,防止超过两次。
public int removeDuplicates(int[] nums) {
int slow = 0;
int fast = 1;
int count = 1;
int max = 2;
for (; fast < nums.length; fast++) {
//当前遍历的数字和慢指针末尾数字不同,就加到慢指针的末尾
if (nums[fast] != nums[slow]) {
slow++;
nums[slow] = nums[fast];
count = 1; //当前数字置为 1 个
//和末尾数字相同,考虑当前数字的个数,小于 2 的话,就加到慢指针的末尾
} else {
if (count < max) {
slow++;
nums[slow] = nums[fast];
count++; //当前数字加 1
}
}
}
return slow + 1;
}
时间复杂度: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)。
总
快慢指针是个好东西,解法二直接利用有序,和倒数第 n 个比,从而保证末尾的数字不超过 n 个,太强了。