删除排序数组中的重复项2

题目:

给定一个增序排列数组 nums ,你需要在 原地 删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。 你不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii

示例 1:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [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]

思路:

  1. 因为不能开辟新数组,所以可以直接在遍历的同时用数组的remove删除重复数字就可以了,但是由于遍历耗时O(n),删除元素需要移动后面的元素,最坏情况耗时O(n^2),所以并不美丽。
  2. 这时我们可以直接用后面的元素替换前面的元素,这样没有删除操作所以耗时为O(n),因为题目说明了超出返回的长度后后面一截就不需要管了

代码:

class Solution {
     
    public int removeDuplicates(int[] nums) {
     
        int i = 0;
        for(int num :nums){
     
            if(i<2 || nums[i-2]<num) nums[i++] = num;
        }
        return i;
    }
}

你可能感兴趣的:(算法一隅,leetcode,java,数组)