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

题意每个数只允许重复一次,输出去重后的数组和个数,依旧 上一题的要求,常数空间也就是只能在原数组上操作。



应为允许一个重复,所以判断就不是相邻而是相隔一个的数字是否不一样。

不一样就需要更新值,但注意到这题更新数组不能是当前的状态,而需要是更新前一个不同的数,用temp存当前值以便于下一次赋值。

因为如果更新当前的数,下一次判断相隔的数是否不同的时候,访问的就是覆盖后的数组值,就不对了。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n==0)return 0;
        if(n==1)return 1;
        int num=1,i,temp=A[1];
        for(i=2;i<n;++i)
            if(A[i]!=A[i-2])
            {
                A[num++]=temp;
                temp=A[i];
            }
        A[num++]=temp;
        return num;
    }
};
// blog.csdn.net/havenoidea


题解目录

你可能感兴趣的:(LeetCode,面试算法)