python 列表去重
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例 1:
输入:nums = [1,1,2] 输出:2, nums = [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] 解释:函数应该返回新的长度 5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
解答
* 题目要求是原地修改
法一
class Solution: def removeDuplicates(self, nums): for i in range(len(nums) - 1, 0, -1): if nums[i] == nums[i - 1]: nums.pop(i) return len(nums), nums s = Solution() nums = [0, 0, 1, 1, 1, 2, 2, 3, 3, 4] length, news_nums = s.removeDuplicates(nums) print(length) print(news_nums)
法二
class Solution: def removeDuplicates(self, nums: List[int]) -> int: p = 0 while (1): if p >= len(nums)-1: break if nums[p] == nums[p+1]: nums.pop(p) else: p += 1 print(nums) Solution().removeDuplicates([1,1,2])
* 不按原地修改
class Solution: def removeDuplicates(self, nums): new_l = [] for i in nums: if i not in new_l: new_l.append(i) return len(new_l), new_l s = Solution() nums = [0, 0, 1, 1, 1, 2, 2, 3, 3, 4] length, news_nums = s.removeDuplicates(nums) print(length) print(news_nums)