标签：
Remove Duplicates from Sorted Array II是Remove Duplicates from Sorted Array的升级版本，即对于一个有序数组，允许最多有两个重复元素。
一个简单直接的思路是维护一个计数器count,用来记录同一个数值元素的出现次数。出现次数为3次即以上，该元素不放入结果的有序数组中。代码如下：
class Solution(object): def removeDuplicates(self, nums): """ :type nums: List[int] :rtype: int """ if not nums: return 0 end = 0 length=len(nums) count = 1 for i in xrange(1,length): temp = nums[i] if temp == nums[end] : count+=1 if count>2: continue end+=1 nums[end]=temp else: count=1 end+=1 nums[end]=temp return end+1
以上代码中，end为当前结果数组的最后一位的index，也即是用来被比较的元素的index。当前元素和cur所代表的数不一样时，count重新置为1，否则加1.算法遍历了一遍数组，时间复杂度为O(n),数组inplace处理，额外空间为count,cur,temp等变量的空间，空间复杂度为O(1).
以上算法没有利用排序数字本身的特性，即连续区间首尾元素相等，则该区间所有元素相等，所以在最多只有两个重复元素时，我们可以直接比较nums[i]和nums[end-1],如果nums[i]=nums[end-1]则nums[i]=nums[end-1]=nums[end],省去了 count的计数操作，代码也大大简化：
class Solution(object): def removeDuplicates(self, nums): """ :type nums: List[int] :rtype: int """ length=len(nums) if length<3: return length end = 1 for i in xrange(2,length): temp = nums[i] if temp != nums[end-1] : end+=1 nums[end]=temp return end+1
时间复杂度O(n),空间复杂度O(1).
Remove Duplicates from Sorted Array II
标签：
原文地址：http://www.cnblogs.com/sherylwang/p/5384563.html
