算法题丨Remove Duplicates from Sorted Array II

描述

Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

示例

Given sorted array nums = [1,1,1,2,2,3],

Your function should return length = 5, with the first five elements of nums 
being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.

算法分析

难度：中
分析：条件参考Remove Duplicates算法，只不过之前元素只允许重复1次，现在改为最多可以重复2次。
思路：既然输入的数组排好序的，那我们定义一个有效元素的数组长度变量i，然后开始遍历元素：
 1. 检索当前元素之前的2个元素，如果跟当前元素相等，说明不满足最多重复2次的条件了，这时候有效元素的数组长度i就不要再自增了；
 2. 否则的话，之前的2个元素，如跟当前元素不相等（小于），说明这个当前元素为有效值，所以将数组末尾元素赋值当前元素,有效元素的数组长度i自增1；
依照上述逻辑循环判断，一直到数组最后一个元素，循环结束后，根据有效元素的数组长度i，获得[0,i)范围内的数组元素，即为题目要求的结果。

代码示例(C#)

public int RemoveDuplicates2(int[] nums)
{
    int i = 0;
    foreach (var num in nums)
    {
        //判断是否有2个以上重复，有的话有效索引+1，并将当前元素赋值到有效数组
        if (i < 2 || num > nums[i - 2])
            nums[i++] = num;
    }
    return i;
}                                           

复杂度

• 时间复杂度：O (n).
• 空间复杂度：O (1).

附录

• 系列目录索引
• 代码实现(C#版)
• 相关算法：Remove Duplicates from Sorted Array

文章作者：原子蛋
文章出处：https://www.cnblogs.com/lizzie-xhu/
个人网站：https://www.lancel0t.cn/
个人博客：https://blog.lancel0t.cn/
微信公众号：原子蛋Live+
扫一扫左侧的二维码（或者长按识别二维码），关注本人微信公共号，获取更多资源。
本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须保留此段声明，且在文章页面明显位置给出原文连接，否则保留追究法律责任的权利。