LeetCode.26 Remove Duplicates from Sorted Array

题目描述和分析

给定有序数组，移除数组中的重复元素使每个元素仅出现一次，并同时返回数组的长度。举一个简单的例子，
$$Input:a1=0,0,1,1,1,2,2,3,3,4$$

$$Return:a2=0,1,2,3,4val=5$$

删除数组中的元素是一项耗时的操作，极端情况下，每次删除数组的头元素，需要进行$\frac{n^2-n}{2}$次移动。

代码

对于这道题，最简单的作法是移除多余的元素，c++中实现起来也比较简单。

int removeDuplicates(vector<int>& nums) {
    unsigned ind = 0;
    
    if (nums.size() == 0)
        return ind;
    
    while (ind < nums.size() - 1) {
        if (nums[ind + 1] == nums[ind])
            nums.erase(nums.begin() + ind + 1);
        else
            ind += 1;
    }
    
    return nums.size();
}

以上做法LeetCode实测耗时为212ms，说明提升的空间非常大。值得注意的是，每次返回的数字可以指定数组选取的位数，即我们可以用交换来代替删除这一操作，交换操作仅需增添$O(1)$的空间代价。具体的代码实现，

int removeDuplicates(vector<int>& nums) {
    unsigned i = 0, j = 1, count = 1;
    int ret;
    
    if (nums.empty())
        return 0;
    
    while (j < nums.size()) {
        if (nums[j] > nums[i]) {
            if (j != i + 1) {
                ret = nums[i + 1];
                nums[i + 1] = nums[j];
                nums[j] = ret;
            } 
            count ++;
            i ++;
        }
        j ++;
    }
    
    return count;
}

LeetCode实测24ms，在时间复杂度上并且达到了top级。总结，在数据结构已被事先固定的情况下，规避耗时的操作是有效的优化方向。