leetcode:Remove Duplicates from Sorted Array II (允许重复一次,去掉数组多余数字)

2014-11-24 00:12:08 · 作者: · 浏览: 4
题目:
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