LeetCode in Python: Remove Duplicates from Sorted Array II

LeetCode in Python

Remove Duplicates from Sorted Array II

class Solution:
    def RemoveDuplicatesFromSortedArrayII(self, nums):
        if len(nums) <= 1:
            return len(nums)
        index = 2
        for i in range(2, len(nums)):
            if nums[i] != nums[index -2]:
                nums[index] = nums[i]
                index = index + 1

        print(nums)
        return index

s = Solution()
nums = [1,1,1,3,4,4,5]
res = s.RemoveDuplicatesFromSortedArrayII(nums)
print(res)