LeetCode算法题:删除排序数组中的重复项II

题目描述如下:

给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

给定 nums = [1,1,1,2,2,3],

函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。

你不需要考虑数组中超出新长度后面的元素。
示例 2:

给定 nums = [0,0,1,1,1,1,2,3,3],

函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。

你不需要考虑数组中超出新长度后面的元素。
说明:

为什么返回数值是整数,但输出的答案是数组呢?

请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

        这道题有两种方法做,一种是暴力法,一种是双指针法,但是这道题很有意思,按照平常的想法,暴力法一般是耗时最长的,但是这道题恰恰相反,在使用相同空间的情况下,双指针法耗费的时间比暴力法长。

        这道题有一个额外的条件,就是不能使用额外空间,而且必须原地修改数组。

        暴力法的思路很简单,先定义一个计数变量,用来记录一个数字出现的次数,如果遇到不同的数字,就使计数变量重新计数;如果遇到相同数字但是计数变量小于2,那么计数变量加1;如果遇到相同数字且计数变量大于等于2,那么从这个位置的下一个位置到数组最后,一次向前进一位。

Python3代码如下:

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        if nums==[]:
            return 0
        lengthOfNums = len(nums)
        number = -10000
        figure = 1
        i = 0
        while i<lengthOfNums:
            if nums[i]!=number:
                figure = 1
                number = nums[i]
            elif nums[i]==number and figure<2:
                figure += 1
            elif nums[i]==number and figure>=2:
                nums[i:lengthOfNums-1] = nums[i+1:lengthOfNums]
                #for j in range(i,lengthOfNums-1):
                #    nums[j] = nums[j+1]
                i = i-1
                lengthOfNums -= 1
            i += 1
        return lengthOfNums

(这里的向前进位,我用了两种方法,一种是Python独有的切片赋值,另一种是老老实实的用for循环赋值,这两种方法的速度差异很大,for循环赋值耗费的时间是切片赋值的三倍左右。)

另一种方法是双指针法。

我们先不考虑边界情况,只考虑中间的情况,假设当前遍历位置为i,写指针的可写入位置为current+1,对于i处的值,其写入的条件是重复小于等于2次,我们考虑已经写入的最后两位current和current-1,这两个位置的情况有两个,相等和不相等,首先考虑相等的情况,此时若i处的值和current-1或者说current处的值相同,那么,i处的值肯定不能加入;然后考虑不相等的情况,即current-1和current处值不相等,那么i处的值无论为什么,都满足题意的,即可以加入,综上所述,当i处的值与current-1处的值不相等时,i处的值可以加入,其他情况均不能加入。
接着考虑边界情况,我们只需要考虑开始即可,开始时,前两个值无论等还是不等,都要原封不动的挪到新数组里,由于新数组就是在原数组上进行修改的,因此前两位直接不动即可,只需要修改遍历指针和写入指针就行。
以上算法只需要进行一次遍历即可,时间复杂度O(n),空间复杂度O(1)。

(这里我引用的是题解的描述,写的很详细)

Python3代码如下:

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        if len(nums)<=2:
            return len(nums)
        
        current = 1
        for i in range(2,len(nums)):
            if nums[i]!=nums[current-1]:
                current = current + 1
                nums[current] = nums[i]
        return current+1

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值