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

【LeetCode】Remove Duplicates from Sorted Array II

时间:2014-05-23 10:23:40      阅读:250      评论:0      收藏:0      [点我收藏+]

标签:style   class   blog   c   code   java   

Remove Duplicates from Sorted Array II

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

For example,
Given sorted array A = [1,1,1,2,2,3],

Your function should return length = 5, and A is now [1,1,2,2,3].

 

我是Pick One!的,所以一开始没关心"Remove Duplicates"中的要求,不知道是排好序的,直接开始做了。

代码比较简单,建立map,存放每个元素与相应的出现次数。

bubuko.com,布布扣
class Solution {
public:
    int removeDuplicates(int A[], int n) 
    {
        if(n <= 2)
            return n;
        int index = 2;
        int iter = 2;
        for(; iter < n; iter ++)
        {
            if(A[iter] != A[index-2])
            {
                A[index] = A[iter];
                index ++;
            }
        }
        return index;
    }
};
bubuko.com,布布扣

bubuko.com,布布扣

 

后来发现是排好序的,那么使用in-place做法就可以了。

使用index记录新的A数组下一个的位置,使用iter扫描原始A数组。

核心思想就是:

如果iter扫到的当前元素在index之前已经存在两个(注意,由于A是排好序的,因此只需要判断前两个就行),

那么iter继续前进。否则将iter指向的元素加入index,index与iter一起前进。

bubuko.com,布布扣
class Solution {
public:
    int removeDuplicates(int A[], int n) 
    {
        if(n <= 2)
            return n;
        int index = 2;
        int iter = 2;
        for(; iter < n; iter ++)
        {
            if(A[iter] != A[index-2])
            {
                A[index] = A[iter];
                index ++;
            }
        }
        return index;
    }
};
bubuko.com,布布扣

bubuko.com,布布扣

【LeetCode】Remove Duplicates from Sorted Array II,布布扣,bubuko.com

【LeetCode】Remove Duplicates from Sorted Array II

标签:style   class   blog   c   code   java   

原文地址:http://www.cnblogs.com/ganganloveu/p/3737536.html

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