一、问题重述
给定一个有序数组nums,请你原地删除重复出现的元素,使每个元素只出现一次,返回删除后数组的长度。不要使用额外的数组空间,必须在原地修改输入数组并在使用O(1)额外空间的条件下完成。
题源来自力扣
提示:以下是本篇文章正文内容
二、思路
因为题目已经给出nums是有序数组,所以其重复元素在数组中的位置是相邻的,所以可以使用两个指针,一个指针用来记录第一次遇到某个元素的下标,再次遇到时,直接跳过,直到遇到新字符,才将其放到第一次遇到的后面。另一个指针用来移动遍历数组。
题目中要求原地修改数组,不得使用额外的数组空间。
因为nums的长度可能为0,所以需要考虑len(nums)==0的特殊情况。
三、代码实现
class Solution():
def removeDuplicate(self, nums):
n = len(nums)
i = 0
if n == 0:
return 0
for j in range(n):
if n[j]!=n[i]:
i+=1
n[j], n[i] = n[i], n[j]
return i+1
四、思考的问题
- 思考问题时,特例需要单独考虑,如本题中,n==0的情况。
- 题目要求要求原地修改数组
- 当判断出不再重复时,需要移动指针i至下一位。
- 返回值也是i+1,因为下标是从0开始的。