【Leetcode】【python】Remove Duplicates from Sorted Array/Remove Duplicates from Sorted Array II

Remove Duplicates from Sorted Array

题目大意

对排好序的list去重，输出去重后长度，并且不能创建新的数组

解题思路

使用一个指针j，当i向后遍历数组时，如果遇到与A[j]不同的，将A[i]和A[j+1]交换，同时j=j+1，即j向后移动一个位置，然后i继续向后遍历。

代码（我提交的）

class Solution(object):    def removeDuplicates(self, nums):        """        :type nums: List[int]        :rtype: int        """        j = 0        for i in range(len(nums)):            if nums[i] == nums[j]:                i += 1                continue            else:                nums[j+1] = nums[i]                i += 1                j += 1        nums = nums[:j+1]        return len(nums)

Remove Duplicates from Sorted Array II

题目大意

在 Remove Duplicates from Sorted Array（从一个有序的数组中去除重复的数字，返回处理后的数组长度） 的基础上，可以使每个数字最多重复一次，也就是说如果某一个数字的个数大于等于2个，结果中应保留2个该数字。

解题思路

参考：http://www.cnblogs.com/zuoyuan/p/3783453.html
使用两个指针prev和curr，判断A[curr]是否和A[prev]、A[prev-1]相等，如果相等curr指针继续向后遍历，直到不相等时，将curr指针指向的值赋值给A[prev+1]。

代码

class Solution(object):    def removeDuplicates(self, nums):        """        :type nums: List[int]        :rtype: int        """        if len(nums) <= 2:             return len(nums)        prev = 1        curr = 2        while curr < len(nums):            # print 'curr:', nums[curr], 'prev, prev-1:', nums[prev], nums[prev - 1]            if nums[curr] == nums[prev] and nums[curr] == nums[prev - 1]:                curr += 1            else:                prev += 1                # print 'fu zhi gei prev hou de yi wei:', nums[prev], nums[curr]                nums[prev] = nums[curr]                curr += 1        return prev + 1

总结

网上答案思路统一，代码不尽相同，python有个set()函数可以用来去重，不过结果一直有误，例如：[1,1,2]输出为[1,1]，没自己研究，比较忙最近，有空的时候好好看一下set()