《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 nums = [1,1,1,2,2,3],Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.

题目大意：将一个已排序的数组中元素出现的次数大于2次的元素的多余部分去除掉。

思路：借助了一个额外数组temp来进行拷贝，最后将temp中的元素全部拷贝回去。

#include<stdio.h>#include<stdlib.h>#include<string.h>int removeDuplicates(int* nums, int numsSize) {    if(nums==NULL||numsSize<1){        return 0;    }    if(numsSize<=2){        return numsSize;    }    //开辟一段空间来进行存储    int *temp=(int *)malloc(numsSize*sizeof(int));    if(temp==NULL){        exit(EXIT_FAILURE);    }    int index1=0;    int index2=0;    int count=1;    int val;     for(int i=1;i<numsSize;i++){        if(nums[i]==nums[index1]){            count++;        }        else{//nums[i]与nums[index]不相等,则将前面的数拷贝到temp数组中进行存储            val=count;            if(count>2){                val=2;            }           for(int j=0;j<val;j++){               temp[index2]=nums[index1+j];               index2++;           }           index1=i;//将index1进行更新           count=1;//将count进行还原，为下一步判断做准备        }    }    //上面没有对最后的进行拷贝    if(index1<numsSize){        val=count;        if(count>2){            val=2;       }        for(int j=0;j<val;j++){               temp[index2]=nums[index1+j];               index2++;           }    }    //再将temp中的数拷贝回去    for(int i=0;i<index2;i++){        printf("%d   ",temp[i]);        nums[i]=temp[i];    }    return index2;}int main(void){    int k;    while(scanf("%d",&k)!=EOF&&k>0){        int *arr=(int *)malloc(k*sizeof(int));        if(arr==NULL){            exit(EXIT_FAILURE);        }        for(int i=0;i<k;i++){            scanf("%d",arr+i);        }        printf("%d\n",removeDuplicates(arr,k));    }}

AC结果如下：

从AC结果可以看出，虽然AC了，但是，效率不高。仍需要改善。

每次AC完自己的代码之后，然后看别人的代码，发现自己真的好渣，虽然最后结果是AC了，但是代码质量没有别人的高

此博客（http://blog.sina.com.cn/s/blog_60b5450101017p33.html）提供了这样一种解法

/*思路：先把序列的前两位放进去，然后让后面的和已经放进去的最后两位比较，如果相同，那么扔掉，如果不同，就放进去。*/int removeDuplicates(int* nums, int numsSize) {    if(nums==NULL||numsSize<1){        return 0;    }    if(numsSize<=2){        return numsSize;    }    int index=2;//始终指向即将要覆盖的位置    for(int i=2;i<numsSize;i++){        if(nums[i]!=nums[index-1]||nums[i]!=nums[index-2]){//跟已经放进去的后两个元素比较，看是否相同            nums[index]=nums[i];            index++;        }    }    return index;}

此代码的运行时间Runtime: 8 ms
但是，从AC结果可以看出，有某种解法只需要 4ms，不知是何种解法。