[牛客网-Leetcode] #数组 中等 remove-duplicates-from-sorted-array

移除重复数字 remove-duplicates-from-sorted-array

题目描述

给定一个已排序的数组，使用就地算法将重复的数字移除，使数组中的每个元素只出现一次，返回新数组的长度。

不能为数组分配额外的空间，你必须使用常数级空间复杂度的就地算法。
例如，
给定输入数组 A=[1,1,2]，
你给出的函数应该返回length=2，A数组现在是[1,2]。

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].

解题思路

• 用快慢两个指针分别用于记录无重复区域和遍历数组
• 指针cnt用于记录无重复区域的右下标，初始为0，代表有一个元素
• 指针index用于遍历数组，初始为1，从第二个元素开始
• 如果A[cnt] == A[index]，则跳过，index++
• 如果A[cnt] ！= A[index]，则A[++cnt] = A[index],index++

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