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());
}