//给你一个有序数组 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 已按升序排列//// Related Topics 数组 双指针// 👍 432 👎 0
移除超过两个相同的项。
因为题目给出的数组是有序的,所以必然相同的元素是相邻的。
双指针
因此可以使用双指针将相同的元素框出来,再移动到数组前列。
如果框出来的元素超过或等于两个,就只移动两个过去,如果不足两个,就只移动一个。
class Solution {public int removeDuplicates(int[] nums) {int n = nums.length;// 标记当前数组前面已经放了多少个元素int index = 0;int i = 0;while (i < n) {// 使用[left, right)去框选相同的元素int left = i;int right = i + 1;// 如果left == right,right就右移while (right < n && nums[right] == nums[left]) {right ++;}// 如果相同的元素数量大于等于2,只移动两个过去if (right - left > 1) {// 有需要删除的数据nums[index] = nums[left];nums[index + 1] = nums[left];index += 2;}else {// 否则说明只有一个元素,只移动一个过去nums[index] = nums[left];index += 1;}// 此时i已经判断完[left, right - 1]i = right;}return index;}}
相同元素计数
因为只要求重复,所以我们只对前两个或者不重复的数字感兴趣,对于什么时候结束重复兴趣不大。
所以可以只记录当前i指示的数字重复了多少次,当小于等于2的时候才移动到数组头。大于2的时候就直接不感兴趣了。
class Solution {public int removeDuplicates(int[] nums) {int i = 1;int count = 1;for (int j = 1; j < nums.length; j++) {// 如果当前数和前面相等的话,计数+1if (nums[j] == nums[j - 1]) {count ++;}else {// 如果不相等,说明换成了新的数字,计数回到1count = 1;}// 如果计数小于等于2,这时移动到数组前部,如果超过2就直接抛弃if (count <= 2) {nums[i] = nums[j];i ++;}}return i;}}
直接移动
如果数组i位置的数相同超过2个,那么到达i的时候 nums[i - 2] == nums[i]
因此可以使用这个特点,不再计数,直接判断是否当前数字已经满了。
class Solution {public int removeDuplicates(int[] nums) {int i = 0;for (int num : nums) {// i < 2是因为,无论数组长什么样,前两个数字肯定是符合要求的// i指示的是当前应该放数字的位置。 如果向前两个位置和当前数字相同,意味着相同数字已经足够了,可以直接抛弃if (i < 2 || nums[i - 2] < num) {nums[i] = num;i ++;}}return i;}}
