【一天一道Leetcode】删除有序数组的重复项Ⅱ

看那个码农

共 2278字,需浏览 5分钟

 ·

2021-04-11 22:50


本篇推文共计2000个字,阅读时间约3分钟。



01


题目描述


题目描述:


给一个有序数组nums,请原地删除重复出现的元素,使每个元素最多出现两次,返回删除后数组的长度。

 

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


示例:

输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5,
并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。
不需要考虑数组中超出新长度后面的元素。


提示:

1.0 <= nums.length <= 3*10^4

2.-10^4 <= nums[i] <= 10^4

3.nums 已按升序排列




02


思路和方法


由题意最后只用输出新数组的长度


我们可以利用遍历,设定两个开始节点。

从数组第一个元素进行遍历,依次对比元素是否相同,将符合题目要求的元素往前移动。


最后输出数组元素不同的那部分长度即可。

 

例如我们用一个例子来直观地说明。

初始数组num=[1,2,2,2]

设定两个起点进行遍历数组,因为题目要求使每个元素最多出现两次,
所以此时我们设置节点的起始值
node=1,j=2,
num[node]=1,num[j]=2,

如果num[i]!=num[node-1]
通过node++,使得node++的值等于此时i的值。
使得num[node]=num[i]

利用这样的方式,将同一个数组的符合要求的元素都往前移动。
最后输出节点node+1的长度即可。




我们用代码表示为:

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





我们由这道题浮想一下,如果题目继续改为:

给一个有序数组nums,请删除重复出现的元素,使每个元素最多出现n次,返回删除后数组的新长度。

 

设定n是一个值,我们的函数表达式可以表示为:

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




往期回顾

【年终总结】你好2021,再见2020。


【秋招纪实录】一篇特别正经的【腾讯】求职经验分享


【一天一道Leetcode】旋转链表



☆ END ☆

你与世界

只差一个

公众号

浏览 46
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报