leetcode-80-删除排序数组中的重复项 II

给定 nums = [1,1,1,2,2,3],

函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。

你不需要考虑数组中超出新长度后面的元素。

给定 nums = [0,0,1,1,1,1,2,3,3],

函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。

你不需要考虑数组中超出新长度后面的元素。

    int removeDuplicates(vector<int>& nums) 
    {
        int i=1,j=0,s1=nums.size(),count=1,t;
        if(s1==0)return 0;//如果vector中没有元素，返回0，不用修改vector
        else if(s1==1)return 1;//如果vector中只有1个元素，返回1，不用修改vector
        while(i<s1)//找到第一个可以被替换的位置，也就是出现了2次以上的元素
        {
            if(nums[i]==nums[i-1])//count初始值为1
            {
                count++;
                if(count>2)//出现了第三次，那么break，当前i的位置就是第一个可以被后面元素替换的位置
                    break;
                i++;
            }
            else//如果不一样了，那么重新初始化count为1
            {
                i++;
                count=1;
            }
        }
        j=i+1;//要找到第一个需要替换的元素位置
        if(i==s1)return i;//边界条件，比如[1,2,3,4],i==s1，那么直接返回i
        if(j==s1)return i;//边界条件，比如[1,1,1],i=2，j=3，j==s1，那么直接返回i
        while(nums[j]==nums[i]&&j<s1)//j还没超出vector的长度，找到第一个不等于nums[i]的元素位置
            j++;
        if(j==s1)return i;//边界条件，如果j==s1都没有找到不等于nums[i]的，那么返回i
        t=nums[j],count=1;//找到了nums[j]!=nums[i]，记录下nums[j]的值，初始化count为1
        nums[i]=nums[j];//修改元素的值
        i++;//到达下一位，i记录的是可以被替换的位置
        j++;//到达下一位，j记录的是需要替换的位置
        while(j<s1)
        {
            if(nums[j]==t)
            {
                count++;
                if(count>2)//如果出现了2次以上，那么我们需要找到跟t不一样的元素值的位置
                {
                    while(nums[j]==t&&j<s1)
                    {
                        j++;
                    }
                    if(j==s1)break;//边界条件，如果没找到，那么break，完全不需要修改vector了
                    t=nums[j];//找到了跟t不一样的元素值，那么更新t
                    nums[i]=nums[j];//修改元素值
                    count=1;//初始化count
                    i++;//i走到下一位
                    j++;//j走到下一位
                }
                else//只出现了2次，那么修改当前nums[i]为nums[j]的值
                {
                    nums[i]=nums[j];
                    i++;
                    j++;
                }
            }
            else//如果跟t不一样，那么修改元素值，更新t值，初始化count
            {
                nums[i]=nums[j];
                count=1;
                t=nums[j];
                i++;
                j++;
            }
        }
        return i;//最后需要返回长度，也就是从0到i的前一位，那么长度为i
    }