【Leetcode】【python】Remove Element

题目大意

去掉数组中等于elem的元素，返回新的数组长度，数组中的元素不必保持原来的顺序。

解题思路

双指针
使用头尾指针，头指针碰到elem时，与尾指针指向的元素交换，将elem都换到数组的末尾去。

代码

判断与指定目标相同

class Solution(object):    def removeElement(self, nums, val):        """        :type nums: List[int]        :type val: int        :rtype: int        """        front = len(nums)-1        behind = len(nums)-1        number = 0        while front >= 0:            print 'now:', front, behind            if nums[front] == val:                print front, behind                number += 1                nums[front], nums[behind] = nums[behind], nums[front]                behind -= 1            front -= 1        print number        return len(nums) - number

判断与指定目标不同

class Solution(object):    def removeElement(self, nums, val):        """        :type nums: List[int]        :type val: int        :rtype: int        """        size = 0        length = len(nums)        for i in range(length):            if nums[i] != val:                nums[size] = nums[i]                size += 1        return size

总结