LeetCode刷题记录【2】(删除数组中重复项Ⅱ)以及删除数组中重复项问题的通解记录

88 阅读3分钟

题目

给你一个有序数组 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 已按升序排列

思路

双指针做即可,和之前的去除重复项的思路都是一样的,抓住是有序数组,相同元素肯定会靠着即可。

代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var removeDuplicates = function(nums) {
    if (nums.length <= 2) {
        return nums.length; // 不可能重复两个以上直接返回掉即可
    }
    // 因为是顺序的数组,所以重复项一定在一起,如果是两个以上的重复项,肯定下标-2还是这个元素
    let slow = 2, fast = 2;  // 双指针,从2开始就行
    while (fast < nums.length) { // 
        if (nums[slow - 2] != nums[fast]) {
            nums[slow] = nums[fast]; // 直接执行覆盖即可
            ++slow;
        }
        ++fast;
    }
    return slow;
};

通解

这种去除有序数组重复项的问题,其实是具备双指针通用解法的:

我们可以发现:对于保留k位的相同数字的话,其实它的具体的方法是这样的:

  • 对于前 k 个数字,我们可以直接保留
  • 对于后面的任意数字,能够保留的前提是:与当前写入的位置前面的第 k 个元素进行比较,不相同则保留。 在这样的前提下,使用双指针形成下面的方法:
/**
 * 保留有序数组的k位重复数字的通解
 * @param {number[]} nums 有序数字数组
 * @param {number} k 保留多少位
 * @returns {number} 新数组的长度
 */
function removeDuplicates(nums, k){
  if(nums.length <= k){ // 长度不允许你重复,直接返回即可
    return nums.length;
  }

  let slow = k, fast = k; // 初始化快慢指针
  while(fast < nums.length){
    // 如果 nums[fast] 不等于 nums[slow - k]
    // 则将 nums[fast] 复制到 nums[slow],并将 slow 向前移动一位
    if (nums[fast] != nums[slow - k]) { //nums[slow - k] 是当前考虑的元素在新数组中的第一个可能的位置
        nums[slow] = nums[fast];
        ++slow;
    }
    // 将 fast 向前移动一位,以检查下一个元素
    ++fast;
  }
  return slow; // 返回新数组长度
}

/**
 * 展示数组用,因为返回的数组仅仅是依靠给定长度来实现展示的,真正的数组中含有无用元素
 * @param {number[]} nums 数组
 * @param {number} length 有效长度
 */
function displayArray(nums, length){
  for(let i = 0; i<length; ++i){
    console.log(nums[i]);
  }
}

let nums = [1,1,1,1,1,2,2,2,3,3,4,5,6,7];
let len = removeDuplicates(nums, 1);
displayArray(nums, len); // 1234567

nums = [1,1,1,1,1,2,2,2,3,3,4,5,6,7];
len = removeDuplicates(nums, 2);
displayArray(nums, len); // 1122334567