Leetcode 26. 删除有序数组中的重复项
题目
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
实例
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
你不需要考虑数组中超出新长度后面的元素。
代码
1 class Solution { 2 public: 3 int removeDuplicates(vector<int>& nums) { 4 int i = 0,j = 1; 5 if(nums.size() == 0 || nums.size() == 1) return nums.size(); 6 while(j < nums.size()){ 7 if(nums[i] == nums[j] ) 8 j++; 9 else{ 10 nums[++i] = nums[j]; 11 j++; 12 } 13 } 14 return i+1; 15 } 16 };
注意
1. 一开始就要考虑边界情况,即元素个数为零或者1时,不需要删除,直接返回元素个数即可
2. j 要小于 nums.size(代表了容器元素的个数)
更新一下
双指针,k 指针保存符合条件的下标,i 来遍历整个数组,符合条件放到 k 位置
1 class Solution { 2 public: 3 int removeDuplicates(vector<int>& nums) { 4 int k = 0; 5 for(int i = 0;i < nums.size();i++){ 6 if(i ==0 || nums[i] != nums[i-1]){ 7 nums[k++] = nums[i]; 8 } 9 } 10 return k; 11 } 12 13 };