码迷,mamicode.com
首页 > 其他好文 > 详细

80. Remove Duplicates from Sorted Array II

时间:2017-08-04 20:38:17      阅读:106      评论:0      收藏:0      [点我收藏+]

标签:mars   code   需要   ice   mos   相同   color   move   维护   

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

For example,
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 doesnt matter what you leave beyond the new length.

原地重置: 此时比较的是新数组的倒数第二个, 即与当前元素相同的元素是否加进去两个了

public int removeDuplicates(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        int index = 0;
        for (int item : nums) {
//---重置原则: 不断挑选的当前元素大于新的数组的倒数第二个数, 如果大于倒数第一个数会少加入重复的数, 新数组变短, 
//如果大于倒数第三个数, 会多加入重复的数 if (index < 2 || item > nums[index - 2]) { nums[index++] = item; } } return index; }

 

构造辅助计数器, 一指示什么时候加元素, 什么时候重置计数器

这道题跟Remove Duplicates from Sorted Array比较类似,区别只是这里元素可以重复出现至多两次,而不是一次。其实也比较简单,只需要维护一个counter,当counter是2时,就直接跳过即可,否则说明元素出现次数没有超,继续放入结果数组,若遇到新元素则重置counter。总体算法只需要扫描一次数组,所以时间上是O(n),空间上只需要维护一个index和counter,所以是O(1)。代码如下: 

 

public int removeDuplicates(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        if (nums.length <= 2) {
            return nums.length;
        }
        int slow = 1, fast = 1, count = 1;
        while (fast < nums.length) {
            if (nums[fast] == nums[fast- 1]) {
                if (count == 0) {
                    fast++;
                } else {
                    count--;
                    nums[slow] = nums[fast];
                    fast++;
                    slow++;
                    
                }
            } else {
                count = 1;
                nums[slow] = nums[fast];
                slow++;
                fast++;
            }
        }
        return slow;
        
        
    }
}

  

80. Remove Duplicates from Sorted Array II

标签:mars   code   需要   ice   mos   相同   color   move   维护   

原文地址:http://www.cnblogs.com/apanda009/p/7286730.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!