Remove Duplicates from Sorted Array II--leetcode 我的java题解

题目是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]. 

费了好长好长时间，脑子总是转不过来。最后终于写出来了。增加了2个变量，一个计数器，一个index标明存放下个不同数据的位置。

如果是没有排序的数组，则需要引入一个hashmap 来记录出现次数。

public class Solution {

    public int removeDuplicates(int[] A) {

     int index=1;

    int samenumber=0;

    if(A.length==0) return 0;

    for(int i=1;i<A.length;i++){

         if(A[i]==A[i-1]) {

             samenumber++;

             if(samenumber==1) {

                 A[index] = A[i];

                 index++;

             }

         }

         if(A[i]!=A[i-1]){

             A[index]=A[i];

             index++;

             samenumber=0;

         }

    }

    return index;

    }

} 

代码虽然长，不过被leetcode接受就可以可以了，扩展性好些。如果这题改成允许每个字符相同最多3次，程序只要改一处地方 if(samenumber==1) 改成if(samenumber<=2) ：

public int removeDuplicates4(int[] A) {
    int index=1;
    int samenumber=0;
    if(A.length==0) return 0;

    for(int i=1;i<A.length;i++){

        if(A[i]==A[i-1]) {
            samenumber++;
            if(samenumber<=2) {
                A[index] = A[i];
                index++;
            }


        }
        if(A[i]!=A[i-1]){
            A[index]=A[i];
            index++;
            samenumber=0;
        }
    }
    return index;
}

加一个别人写的代码c++的代码

// LeetCode, Remove Duplicates from Sorted Array II
// 时间复杂度O(n)，空间复杂度O(1)
// @author hex108 (https://github.com/hex108)

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