LintCode（101）删除排序数组中的重复数字 II

Posted 2020-07-19 逆風的薔薇

题目

跟进“删除重复数字”：

如果可以允许出现两次重复将如何处理？

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出数组A =[1,1,1,2,2,3]，你的函数应该返回长度5，此时A=[1,1,2,2,3]。

分析

与上题思路相同，只需要增加一个记录元素出现次数的变量即可，限制最多出现2次。

Python代码

class Solution:
    """
    @param A: a list of integers
    @return an integer
    """
    def removeDuplicates(self, A):
        # write your code here
        if len(A) == 0:
            return 0
            
        times = 1
        k = 0
        for i in range(1,len(A)):
            if A[i] != A[k]:
                k += 1
                A[k] = A[i]
                times = 1
            else:
                if times >= 2:
                    continue
                else:
                    k += 1
                    A[k] = A[i]
                    times += 1
        
        del A[k+1:len(A)]
        
        return len(A)
          

GitHub -- Python代码

C++代码

/*
101 删除排序数组中的重复数字 II

跟进“删除重复数字”：

如果可以允许出现两次重复将如何处理？



您在真实的面试中是否遇到过这个题？ Yes
样例
给出数组A =[1,1,1,2,2,3]，你的函数应该返回长度5，此时A=[1,1,2,2,3]。

*/

class Solution {
public:
    /**
     * @param A: a list of integers
     * @return : return an integer
     */
    int removeDuplicates(vector<int> &nums) {
        // write your code here
        if(nums.empty())
        {
            return 0;
        }//if
        
        int n = nums.size(), k=0, times=1;
        for(int i=1; i<n; ++i)
        {
            if(nums[i] != nums[k])
            {
                nums[++k] = nums[i];
                times = 1;
            }else if(nums[i] == nums[k]){
                if(times >= 2)
                {
                    continue;
                }else{
                    nums[++k] = nums[i];
                    ++times;
                }//else
            }//elif
        }//for
        
        nums.resize(k+1);
        return k+1;
    }
};

GitHub -- C++代码