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

题目

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。
不要使用额外的数组空间，你必须在原地修改输入数组 并在使用 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。 你不需要考虑数组中超出新长度后面的元素。

代码

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

双下标法，i下标指向旧的有重复元素的nums，j下表指向新的没有重复元素的nums。当nums[i]与nums[j]不相同时，在nums[j]中添加nums[i]，j向后移动。最后j的长度便是nums的新长度。
小技巧：可以通过减少vector函数的调用来降低运行时间，为此创建了length变量。

结果

另外给出两种STL的解法，不过效率都不太好

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        return distance(nums.begin(),removeDuplicates(nums.begin(),nums.end(),nums.begin()));
    }
    template<typename InIt,typename OutIt>
    OutIt removeDuplicates(InIt first,InIt last,OutIt output)
    {
        while(first!=last)
        {
            *output++ = *first;
            first = upper_bound(first,last,*first);
        }
        return output;
    }
};

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        return distance(nums.begin(),unique(nums.begin(),nums.end()));
    }
};

参考资源：

Leetcode 26
Leetcode题解