题目:
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。 不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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]); }
解题思路:本题比较简单。用两个指针即可完成。一个为pre指针记录删除新的数组,另一个为cur指针,指向当前的元素。如果发现前一个数据与后一数据不想等,这时将前一个数据添加到新的数组中;
python:
class Solution: def removeDuplicates(self, nums): """ :type nums: List[int] :rtype: int """ length = len(nums) if length <= 1: return length pre = 0 cur = 1 while cur < length: while cur < length and nums[cur] == nums[pre]: cur += 1 if cur == length: break else: pre += 1 nums[pre] = nums[cur] cur += 1 return pre+1
c++:
class Solution { public: int removeDuplicates(vector<int>& nums) { if(nums.size() <= 1){ return nums.size(); } int pre = 0; int cur = 1; while(cur < nums.size()){ while(nums[cur] == nums[pre] && cur < nums.size()){ cur++; } if(cur == nums.size()){ break; }else{ pre++; nums[pre] = nums[cur]; } cur++; } return pre+1; } };