题目:
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