80. remove duplicates from sorted array ii

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

        i = 0
        for n in nums:
            if i < 2 or n > nums[i - 2]:
                nums[i] = n
                i += 1
        return i
if __name__ == "__main__":
    answer=Solution()
    print answer.removeDuplicates([1,1,1,2,2,3])