首页 > 代码库 > 【LeetCode】Remove Duplicates from Sorted Array II

【LeetCode】Remove Duplicates from Sorted Array II

2024-07-05 03:19:27   218人阅读

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，存放每个元素与相应的出现次数。

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;
    }
};

 

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

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

核心思想就是：

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

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

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;
    }
};


联系
我们