Leetcode(27)-- 移除元素

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度5,并且 nums 中的前五个元素为0,1,3,0,4

---

方法一：直接使用删除的方法（简单版）

class Solution:
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        l = len(nums)
        i = 0
        while i<l:
            if nums[i]==val:
                nums.remove(nums[i])
                l-=1
            else:
                i+=1
        return len(nums)

小结：注意这里使用的列表如果使用删除的话，列表的索引会自动跟着变化的， 从删除的位置之后所有的元素会整体向前移，所以这点是关键，需要注意，给个例子，你就会明白。

li = [1,2,2,3,4,5,6]
for i in range(len(li)):
    if li[i]==2:
        li.remove(li[i])
    print(li[i])


Traceback (most recent call last):
1
2
3
4
5
6
  File "E:/python代码文件/作业/Tensorflow/Tersorflow-1.py", line 7, in <module>
    if li[i]==2:
IndexError: list index out of range

Process finished with exit code 1

        从这里可以看出，当删除第一个2之后，第二个2会向前移动，索引也会跟着变化，第二个的2的索引从开始的2变成后来的1（索引），所以你会看到还有2输出（我们的本意是删除列表中所有2，有没有发现，当我们用列表时，使用删除的操作的时候，要格外注意这点），当然len(li)也就变化了吗。即从删除的地方开始所有后面的元素会整体向前移动。

第二种方法：双指针的方法

这个和前面的一题极其的相似，可以对照着看下。

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