Leetcode 26. 删除排序数组中的重复项

题目

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

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

实例

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

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

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

代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int i = 0,j = 1; 
        if(nums.size() == 0 || nums.size() == 1)  return nums.size();
        while(j < nums.size()){
            if(nums[i] == nums[j] )
                j++;
            else{
                nums[++i] = nums[j];
                j++;
            }         
        }
        return i+1;
    }
};

注意

1. 一开始就要考虑边界情况，即元素个数为零或者1时，不需要删除，直接返回元素个数即可

2. j 要小于 nums.size(代表了容器元素的个数)