删除有序数组的重复元素

int removeDuplicates(vector<int>& nums) {
    if (nums.empty()){
        return 0;
    }
    
    auto it = nums.begin() + 1;
    int currentVal = nums.front();
    while(it != nums.end()){
        if (*it == currentVal){
            it = nums.erase(it);
        }
        else{
            currentVal = *it;
            ++it;
        }
    }
    return static_cast<int>(nums.size());
}

 以上是我第一次的代码,用了 80 ms,很慢啊。经过 profile 之后,我发现时间瓶颈是内存的频繁获取与释放。所以,修改之后的代码是这样的,32 ms。

int removeDuplicates(vector<int>& nums) {
    if (nums.empty()){
        return 0;
    }
    
    vector<int> result{nums.front()};
    
    auto it = nums.begin() + 1;
    while(it != nums.end()){
        if (*it != *(it - 1)){
            result.push_back(*it);
        }
        ++it;
    }
    nums = move(result);
    return static_cast<int>(nums.size());
}

 

posted @ 2015-08-22 18:32  wu_overflow  阅读(176)  评论(0编辑  收藏  举报