LeetCode 26. Remove Duplicates from Sorted Array

1. 题目描述
   Given a sorted array nums, remove the duplicates in-place such that each element appear only once and return the new length.
   Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.
2. 解决方案
   题目要求不借助外部空间，直接在原数组上进行操作，删除其中重复的元素。因为题目中已经给出是排好序的数组，所以重复元素是相连的，这里用Current_num记录第一个出现的元素（不管重复与否），判断后续元素是否与其相等，如果相等，则删除后续元素；否则，说明该元素只出现一次，不需要删除，则将Current_num换成下一个元素。
3. 代码

#include <iostream>
#include <vector>
using namespace std;
int removeDuplicates(vector<int>& nums);

int main()
{
    vector<int> vec;
    int a[] = {0,0,1,1,1,2,2,3,3,4};
    for (int i=0; i<sizeof(a)/sizeof(a[0]); i++){
        vec.push_back(a[i]);
    }
    int result = removeDuplicates(vec);
    cout << result << endl;
    return 0;
}

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