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

题目描述:

给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

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

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

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

示例 2:

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

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

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

 

要完成的函数:

int removeDuplicates(vector<int>& nums) 

 

说明:

1、这道题给定一个vector,里面存放着升序排列的int型整数。

这些整数可能有多个,比如[1,1,1,2,2,3,4,5,5,5,5,5],要求每个整数只能出现2次,所以最终变成[1,1,2,2,3,4,5,5]。

为此你需要原地修改vector的数值,要求只能使用O(1)的空间。

最后返回新的vector的长度,不需要返回修改完的vector,但是vector要原地修改完成。

不需要在意修改完的vector后面的元素,测试样例在测试时只会在意你返回的长度以内的vector元素。

 

2、这道题从思路上来说,一点也不难,只是比较繁琐。

我们看一下举的这个例子,[1,1,1,2,2,2,3,3,3]

我们需要先找到第一个可以被替换的位置,第三个1,也就是位置是2。

然后找到第一个需要更换的位置,第一个2,也就是位置是3。

把第一个2放在第三个1那里。

接着继续往后面走,继续匹配。第二个2要放在第一个2的位置上,接着第三个2由于已经出现了两次了,所以不理。

到达下一个不一样的元素3,把3放在第二个2的位置上,把第二个3放在第三个2的位置上,最后的第三个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
    }

上述代码实测12ms,beats 99.85% of cpp submissions。

这道题其实思路很简单,就是琐碎了点,需要测试各种边界条件。

posted @ 2018-08-22 13:11  chenjx85  阅读(255)  评论(0编辑  收藏  举报