删除有序数组中的重复元素 Remove Duplicates from Sorted Array

最新推荐文章于 2024-01-02 23:35:28 发布

原创最新推荐文章于 2024-01-02 23:35:28 发布 · 1.8k 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 3 个专栏收录

151 篇文章

订阅专栏

我爱算法

142 篇文章

订阅专栏

字符串问题

45 篇文章

订阅专栏

本文介绍了一种不使用额外空间去除已排序数组中重复元素的方法，通过双指针技巧实现原地修改，确保每个元素只出现一次，并返回新的长度。提供了两种实现方案，一种使用 while 循环，另一种采用 for 循环。

题目源自于Leetcode。

题目：Given a sorted array, remove the duplicates in place such that each element appear only once and return the new length.

Do not allocate extra space for another array, you must do this in place with constant memory.

For example,
Given input array A = [1,1,2],
Your function should return length = 2, and A is now [1,2].

思路：左/右两个指针控制着原地的新/旧数组。

注意时刻关注着右指针是否到达末尾。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n<=1)
            return n;
        int left, right;
        left = 0;
        right = 1;
        while(right<n)
        {
            while(right<n && A[right] == A[left])
                right++;
            if(right == n)
                break;

            left++;
            A[left] = A[right];
            right++;
        }
        return left+1;
    }
};

或者用for循环也一样。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n == 0 || A == NULL)
            return 0;
            
        int index = 0;
        for(int i=1;i<n;i++)
        {
            if(A[index] != A[i])
            {
                ++index;
                A[index] = A[i];
            }
        }
        return index+1;
    }
};

如果想偷懒使用STL算法的话，可以先unique()将重复元素仍在后面，然后再返回非重复元素的个数。但是这样的实现效率并没有直接双指针快。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        return distance(A, unique(A, A+n));
    }
};