Leetcode_remove-duplicates-from-sorted-array-ii (c++ and python version)

最新推荐文章于 2020-06-20 11:22:21 发布

原创最新推荐文章于 2020-06-20 11:22:21 发布 · 763 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 同时被 3 个专栏收录

155 篇文章

订阅专栏

c++

62 篇文章

订阅专栏

python

52 篇文章

订阅专栏

本文介绍了一种算法，用于从已排序的数组中移除重复元素，最多允许保留两个重复项。提供了两种解决方案，一种使用map进行标记，另一种采用更高效的方法，通过追踪当前有效数组长度和重复标志来实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

地址：http://oj.leetcode.com/problems/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].

思路：笔者用的map来标记的。应该可以不用map直接用几个下标来标记，这样的话额外空间就小了。因为题目说好了数组是排好序的，用map的话就没用利用这一点。

参考代码：

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        map<int, int>dm;
        for(int i = 0; i < n; ++i)
        {
            if(dm[A[i]]==0)
            {
                dm[A[i]] = 1;
            }
            else if(dm[A[i]]==1)
            {
                dm[A[i]] = 2;
            }
            else
            {
                for(int j = i; j < n-1; ++j)
                {
                    A[j] = A[j+1];
                }
                --i;
                --n;
            }
        }
        return n;
    }
};

//SECOND TRIAL, in place
class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n<=2)
            return n;
        int len = 1;
        bool flag = true;
        for(int i = 1; i<n; ++i)
        {
            if(A[i]!=A[len-1])
            {
                A[len++] = A[i];
                flag = true;
            }
            else if(flag)
            {
                A[len++] = A[i];
                flag = false;
            }
        }
        return len;
    }
};

python:

class Solution:
    # @param A a list of integers
    # @return an integer
    def removeDuplicates(self, A):
        if len(A)<=2:
            return len(A)
        cnt = 1
        flag = True
        for i in range(1, len(A)):
            if A[i] != A[i-1]:
                flag = True
                A[cnt] = A[i]
                cnt += 1
            elif flag:
                flag = False
                A[cnt] = A[i]
                cnt += 1
        return cnt