【LEETCODE】80-Remove Duplicates from Sorted Array II [Python]

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.

题意：
接Remove Duplicates
如果允许重复的数字最多出现两次呢？
例：
给一个数组：nums = [1,1,1,2,2,3]
返回length = 5, 即 1, 1, 2, 2 and 3

思路：
两个指针 pre ， cur
A[cur] 如果和 A[pre]及 A[pre-1]的值想等，则 cur进1，一直到 A[cur] 不再等于 A[pre]及 A[pre-1]
当不是上述情况时，pre进1， A[pre]换成 A[cur] ，cur进1，也就是把多余两个的数字，用后面的非重复数字替换掉



Python:

class Solution(object):
def removeDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: int
"""

n=len(nums)
if n<=2:
return n

pre=1
cur=2

while cur<n:

if nums[cur]==nums[pre] and nums[cur]==nums[pre-1]:
cur+=1
else:
pre+=1
nums[pre]=nums[cur]
cur+=1
return pre+1

错误分析：







第二个图，如果遇到多余3个相等的数字的时候，就会给出错误答案
第一个图，用一个 while 把第二个图的错误改过来了，但是结果数组的顺序又被搞乱了，虽然要求是返回 length，但是内部会检测是否保证原有大小顺序