从排序数组中删除重复项

37次阅读
没有评论

题目描述

题目来自leetcode https://leetcode-cn.com/explore/interview/card/top-interview-questions-easy/1/array/21/

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

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

示例 1:

给定数组 nums = [1,1,2], 

函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。 

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

示例 2:

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

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

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

说明:

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

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

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

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

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

语言:python3

解法1: 单指针

比较相邻值,相等则删除前一个值,指针不后移,不等指针后移,循环比较

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        i = 0
        while i < len(nums)-1:
            if nums[i] == nums[i+1]:
              # 删除前一个值,保证nums[i]是重复值的第二个值,这样在遍历时顺序不会乱
                nums.remove(nums[i])  
            else:
              # 不相等则指针后移
                i += 1
        return len(nums)

解法2: 双指针(效率最高)

首先注意数组是有序的,那么重复的元素一定会相邻。

要求删除重复元素,实际上就是将不重复的元素移到数组的左侧。

考虑用 2 个指针,一个在前记作 j,一个在后记作 k,算法流程如下:

1.比较 j 和 k 位置的元素是否相等。

如果相等,j 后移 1 位
如果不相等,将 j 位置的元素复制到 k+1 位置上,j 后移一位,k 后移 1 位
这么做的目的是保证数组的第一个元素到k指针所在的元素为已排序且不重复
重复上述过程,直到 j 等于数组长度。

返回 k + 1,即为新数组长度。

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        if not nums or len(nums) == 0:
            return 0
        j, k = 0, 0
        while j < len(nums):
            if nums[k] != nums[j]:
                nums[k+1] = nums[j]
                k += 1
            j += 1
        return k + 1

解法3: 利用python自建函数(效率最低)

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
      # 去重,重新排序,转化为python list, 再赋值给数组
        nums[:] = list(sorted(set(nums))) 
        return len(nums)
正文完
 
dustin
版权声明:本站原创文章,由 dustin 2019-12-12发表,共计1395字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)
验证码