#100 Remove Duplicates from Sorted Array

最新推荐文章于 2022-02-11 13:21:04 发布

原创最新推荐文章于 2022-02-11 13:21:04 发布 · 268 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode #two pointers

lintcode 专栏收录该内容

217 篇文章

订阅专栏

本文介绍了一种使用双指针技术去除已排序数组中重复元素的方法，该方法可以在常数空间复杂度内完成操作，适用于不希望额外分配内存的情况。

题目描述：

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.

Have you met this question in a real interview?

Yes

Example

Given input array A = [1,1,2],

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

题目思路：

这题还是用two pointers，一个i去遍历，一个idx守在原地。如果A[i] == A[idx]，就不取A[i];否则，A[idx + 1] = A[i]。

Mycode（AC = 42ms）：

class Solution {
public:
    /**
     * @param A: a list of integers
     * @return : return an integer
     */
    int removeDuplicates(vector<int> &nums) {
        // write your code here
        if (nums.size() <= 1) return nums.size();
        
        int idx = 0;
        for (int i = 1; i < nums.size(); i++) {
            if (nums[i] != nums[idx]) {
                nums[idx + 1] = nums[i];
                idx++;
            }
        }
        
        return idx + 1;
    }
};